HomeSort by relevance Sort by last modified time
    Searched refs:AddDim (Results 1 - 25 of 112) sorted by null

1 2 3 4 5

  /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
k_feature_gradient_op.cc 113 out_routes_shape.AddDim(num_data);
114 out_routes_shape.AddDim(num_nodes);
118 out_data_shape.AddDim(num_nodes);
119 out_data_shape.AddDim(num_features);
123 out_weights_shape.AddDim(num_data);
124 out_weights_shape.AddDim(num_nodes);
125 out_weights_shape.AddDim(num_features_per_node);
stochastic_hard_routing_gradient_op.cc 134 output_routing_shape.AddDim(num_data);
135 output_routing_shape.AddDim(num_nodes);
139 output_data_shape.AddDim(num_nodes);
140 output_data_shape.AddDim(num_features);
144 output_parameters_shape.AddDim(num_data);
145 output_parameters_shape.AddDim(num_nodes);
146 output_parameters_shape.AddDim(num_features);
150 output_bias_shape.AddDim(num_data);
hard_routing_function_op.cc 116 output_probability_shape.AddDim(num_data);
117 output_probability_shape.AddDim(tree_depth_);
121 output_path_shape.AddDim(num_data);
122 output_path_shape.AddDim(tree_depth_);
stochastic_hard_routing_function_op.cc 129 output_probability_shape.AddDim(num_data);
130 output_probability_shape.AddDim(tree_depth_);
134 output_path_shape.AddDim(num_data);
135 output_path_shape.AddDim(tree_depth_);
unpack_path_op.cc 86 output_shape.AddDim(num_data);
87 output_shape.AddDim(num_nodes);
routing_function_op.cc 107 output_shape.AddDim(num_data);
108 output_shape.AddDim(max_nodes_);
routing_gradient_op.cc 106 output_shape.AddDim(num_data);
107 output_shape.AddDim(max_nodes_);
k_feature_routing_function_op.cc 122 output_shape.AddDim(num_data);
123 output_shape.AddDim(max_nodes_);
  /external/tensorflow/tensorflow/core/kernels/
batchtospace_op.cc 137 external_output_shape.AddDim(orig_input_batch_size / block_shape_product);
143 external_output_shape.AddDim(size);
145 internal_input_shape.AddDim(input_batch_size);
146 internal_output_shape.AddDim(input_batch_size / block_shape_product);
161 internal_input_shape.AddDim(input_size);
162 internal_output_shape.AddDim(cropped_size);
163 external_output_shape.AddDim(cropped_size);
170 external_output_shape.AddDim(size);
173 internal_input_shape.AddDim(depth);
174 internal_output_shape.AddDim(depth)
    [all...]
spacetobatch_op.cc 134 external_output_shape.AddDim(orig_input_tensor.dim_size(0) *
141 external_output_shape.AddDim(size);
143 internal_input_shape.AddDim(input_batch_size);
144 internal_output_shape.AddDim(input_batch_size * block_shape_product);
162 internal_input_shape.AddDim(input_size);
164 internal_output_shape.AddDim(output_size);
165 external_output_shape.AddDim(output_size);
172 external_output_shape.AddDim(size);
175 internal_input_shape.AddDim(depth);
176 internal_output_shape.AddDim(depth)
    [all...]
svd_op_gpu.cu.cc 113 u_shape.AddDim(p);
114 u_shape.AddDim(m);
245 input_shape.AddDim(n);
246 input_shape.AddDim(m);
315 shapeS.AddDim(p);
318 shapeU.AddDim(m);
319 shapeU.AddDim(m);
320 shapeV.AddDim(n);
321 shapeV.AddDim(n);
323 shapeU.AddDim(m)
    [all...]
reduction_ops_common.cc 24 for (auto size : out_reshape_) shape.AddDim(size);
31 for (auto size : out_shape_) shape.AddDim(size);
39 shape.AddDim(data_reshape_[i]);
42 shape.AddDim(data_reshape_[i]);
decode_raw_op.cc 54 out_shape.AddDim(0);
66 out_shape.AddDim(added_dim);
reshape_op.h 109 shape->AddDim(1);
114 shape->AddDim(size);
gather_op.cc 93 result_shape.AddDim(params.dim_size(i));
98 result_shape.AddDim(params.dim_size(i));
reshape_util.cc 71 output_shape.AddDim(1);
77 output_shape.AddDim(size);
diag_op.cc 55 out_shape.AddDim(diagonal.dim_size(i));
58 out_shape.AddDim(diagonal.dim_size(i));
95 out_shape.AddDim(tensor.dim_size(i));
matrix_diag_op.cc 63 output_shape.AddDim(input_shape.dim_size(i));
67 output_shape.AddDim(min_dim);
104 output_shape.AddDim(k);
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
reshape_op.cc 62 shape.AddDim(1);
67 shape.AddDim(size);
matrix_set_diag_op.cc 50 expected_diag_shape.AddDim(min_dim);
  /external/tensorflow/tensorflow/core/framework/
tensor_shape_test.cc 67 s.AddDim(100);
179 s.AddDim(1);
181 s.AddDim(100000);
184 s.AddDim(2);
186 s.AddDim(4);
188 s.AddDim(3);
228 /// then call `AddDim()` on.
243 void AddDim(int64 size);
386 AddDim(d.size());
394 AddDim(s)
    [all...]
  /external/tensorflow/tensorflow/core/util/
strided_slice_op.cc 266 processing_shape->AddDim(shrink_i ? 1 : -1);
353 processing_shape->AddDim(size_i);
355 processing_shape->AddDim(-1);
367 final_shape->AddDim(processing_shape->dim_size(gather_index));
369 final_shape->AddDim(1);
  /external/tensorflow/tensorflow/contrib/periodic_resample/ops/
array_ops.cc 57 result_shape.AddDim(target_dimensions[i]);
  /external/tensorflow/tensorflow/lite/tools/accuracy/
utils.cc 55 shape.AddDim(tflite_tensor.dims->data[i]);
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_resource.cc 137 ta_shape.AddDim(max_array_size_);
145 ta_shape.AddDim(max_array_size_);
170 ta_shape.AddDim(max_array_size_);

Completed in 465 milliseconds

1 2 3 4 5