Home
last modified time | relevance | path

Searched refs:AddDim (Results 1 – 25 of 112) sorted by relevance

12345

/external/tensorflow/tensorflow/core/kernels/
Dbatchtospace_op.cc137 external_output_shape.AddDim(orig_input_batch_size / block_shape_product); in BatchToSpaceOpCompute()
143 external_output_shape.AddDim(size); in BatchToSpaceOpCompute()
145 internal_input_shape.AddDim(input_batch_size); in BatchToSpaceOpCompute()
146 internal_output_shape.AddDim(input_batch_size / block_shape_product); in BatchToSpaceOpCompute()
161 internal_input_shape.AddDim(input_size); in BatchToSpaceOpCompute()
162 internal_output_shape.AddDim(cropped_size); in BatchToSpaceOpCompute()
163 external_output_shape.AddDim(cropped_size); in BatchToSpaceOpCompute()
170 external_output_shape.AddDim(size); in BatchToSpaceOpCompute()
173 internal_input_shape.AddDim(depth); in BatchToSpaceOpCompute()
174 internal_output_shape.AddDim(depth); in BatchToSpaceOpCompute()
Dspacetobatch_op.cc134 external_output_shape.AddDim(orig_input_tensor.dim_size(0) * in SpaceToBatchOpCompute()
141 external_output_shape.AddDim(size); in SpaceToBatchOpCompute()
143 internal_input_shape.AddDim(input_batch_size); in SpaceToBatchOpCompute()
144 internal_output_shape.AddDim(input_batch_size * block_shape_product); in SpaceToBatchOpCompute()
162 internal_input_shape.AddDim(input_size); in SpaceToBatchOpCompute()
164 internal_output_shape.AddDim(output_size); in SpaceToBatchOpCompute()
165 external_output_shape.AddDim(output_size); in SpaceToBatchOpCompute()
172 external_output_shape.AddDim(size); in SpaceToBatchOpCompute()
175 internal_input_shape.AddDim(depth); in SpaceToBatchOpCompute()
176 internal_output_shape.AddDim(depth); in SpaceToBatchOpCompute()
Dsvd_op_gpu.cu.cc113 u_shape.AddDim(p); in RunSVD()
114 u_shape.AddDim(m); in RunSVD()
245 input_shape.AddDim(n); in PerformSVD_MgeqN()
246 input_shape.AddDim(m); in PerformSVD_MgeqN()
315 shapeS.AddDim(p); in ComputeAsync()
318 shapeU.AddDim(m); in ComputeAsync()
319 shapeU.AddDim(m); in ComputeAsync()
320 shapeV.AddDim(n); in ComputeAsync()
321 shapeV.AddDim(n); in ComputeAsync()
323 shapeU.AddDim(m); in ComputeAsync()
[all …]
Dreduction_ops_common.cc24 for (auto size : out_reshape_) shape.AddDim(size); in out_reshape()
31 for (auto size : out_shape_) shape.AddDim(size); in out_shape()
39 shape.AddDim(data_reshape_[i]); in shuffled_shape()
42 shape.AddDim(data_reshape_[i]); in shuffled_shape()
Dmatrix_diag_op.cc63 output_shape.AddDim(input_shape.dim_size(i)); in Compute()
67 output_shape.AddDim(min_dim); in Compute()
104 output_shape.AddDim(k); in Compute()
Dreshape_op.h109 shape->AddDim(1); in ValidateSizes()
114 shape->AddDim(size); in ValidateSizes()
Ddiag_op.cc55 out_shape.AddDim(diagonal.dim_size(i)); in Compute()
58 out_shape.AddDim(diagonal.dim_size(i)); in Compute()
95 out_shape.AddDim(tensor.dim_size(i)); in Compute()
Ddecode_raw_op.cc54 out_shape.AddDim(0); in Compute()
66 out_shape.AddDim(added_dim); in Compute()
Dmkl_batch_matmul_op.cc75 out_shape.AddDim(lhs.dim_size(i)); in Compute()
89 out_shape.AddDim(lhs_rows); in Compute()
90 out_shape.AddDim(rhs_cols); in Compute()
/external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/core/ops/
Dk_feature_gradient_op.cc113 out_routes_shape.AddDim(num_data); in Compute()
114 out_routes_shape.AddDim(num_nodes); in Compute()
118 out_data_shape.AddDim(num_nodes); in Compute()
119 out_data_shape.AddDim(num_features); in Compute()
123 out_weights_shape.AddDim(num_data); in Compute()
124 out_weights_shape.AddDim(num_nodes); in Compute()
125 out_weights_shape.AddDim(num_features_per_node); in Compute()
Dstochastic_hard_routing_gradient_op.cc134 output_routing_shape.AddDim(num_data); in Compute()
135 output_routing_shape.AddDim(num_nodes); in Compute()
139 output_data_shape.AddDim(num_nodes); in Compute()
140 output_data_shape.AddDim(num_features); in Compute()
144 output_parameters_shape.AddDim(num_data); in Compute()
145 output_parameters_shape.AddDim(num_nodes); in Compute()
146 output_parameters_shape.AddDim(num_features); in Compute()
150 output_bias_shape.AddDim(num_data); in Compute()
Dhard_routing_function_op.cc116 output_probability_shape.AddDim(num_data); in Compute()
117 output_probability_shape.AddDim(tree_depth_); in Compute()
121 output_path_shape.AddDim(num_data); in Compute()
122 output_path_shape.AddDim(tree_depth_); in Compute()
Dstochastic_hard_routing_function_op.cc129 output_probability_shape.AddDim(num_data); in Compute()
130 output_probability_shape.AddDim(tree_depth_); in Compute()
134 output_path_shape.AddDim(num_data); in Compute()
135 output_path_shape.AddDim(tree_depth_); in Compute()
Dunpack_path_op.cc86 output_shape.AddDim(num_data); in Compute()
87 output_shape.AddDim(num_nodes); in Compute()
Drouting_function_op.cc107 output_shape.AddDim(num_data); in Compute()
108 output_shape.AddDim(max_nodes_); in Compute()
Drouting_gradient_op.cc106 output_shape.AddDim(num_data); in Compute()
107 output_shape.AddDim(max_nodes_); in Compute()
Dk_feature_routing_function_op.cc122 output_shape.AddDim(num_data); in Compute()
123 output_shape.AddDim(max_nodes_); in Compute()
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape_test.cc67 s.AddDim(100); in TEST()
179 s.AddDim(1); in TEST()
181 s.AddDim(100000); in TEST()
184 s.AddDim(2); in TEST()
186 s.AddDim(4); in TEST()
188 s.AddDim(3); in TEST()
243 void AddDim(int64 size);
386 AddDim(d.size()); in TensorShapeOld()
394 AddDim(s); in TensorShapeOld()
405 void TensorShapeOld::AddDim(int64 size) { in AddDim() function in tensorflow::__anone850e9110111::TensorShapeOld
[all …]
Dtensor_shape.cc130 AddDim(d.size()); in TensorShapeBase()
225 AddDim(internal::SubtleMustCopy(s)); in InitDims()
323 void TensorShapeBase<Shape>::AddDim(int64 size) { in AddDim() function in tensorflow::TensorShapeBase
384 for (auto d : shape) AddDim(d.size); in AppendShape()
398 AddDim(dval); in InsertDim()
431 AddDim(dval); in set_dim()
452 AddDim(dval); in RemoveDimRange()
636 out.AddDim(size); in Concatenate()
646 for (auto dim : shape) out.AddDim(dim.size); in Concatenate()
676 result->AddDim(dim0 >= 0 ? dim0 : dim1); in MergeWith()
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc266 processing_shape->AddDim(shrink_i ? 1 : -1); in ValidateStridedSliceOp()
353 processing_shape->AddDim(size_i); in ValidateStridedSliceOp()
355 processing_shape->AddDim(-1); in ValidateStridedSliceOp()
367 final_shape->AddDim(processing_shape->dim_size(gather_index)); in ValidateStridedSliceOp()
369 final_shape->AddDim(1); in ValidateStridedSliceOp()
Dexample_proto_fast_parsing.cc1002 out_shape.AddDim(serialized.size()); in FastParseExample()
1004 out_shape.AddDim(dim); in FastParseExample()
1096 indices_shape.AddDim(total_num_features); in FastParseExample()
1097 indices_shape.AddDim(2); in FastParseExample()
1102 values_shape.AddDim(total_num_features); in FastParseExample()
1179 values_shape.AddDim(batch_size); in FastParseExample()
1180 values_shape.AddDim(max_num_elements); in FastParseExample()
1182 values_shape.AddDim(config.dense[d].shape.dim_size(i)); in FastParseExample()
1485 values_shape.AddDim(num_elements / num_elements_divisor); in FastParseSingleExample()
1487 values_shape.AddDim(config.dense[d].shape.dim_size(i)); in FastParseSingleExample()
[all …]
Dexample_proto_helper.cc310 output_shapes->indices_shape.AddDim(total_num_features); in GetSparseTensorShapes()
311 output_shapes->indices_shape.AddDim(2); in GetSparseTensorShapes()
312 output_shapes->values_shape.AddDim(total_num_features); in GetSparseTensorShapes()
346 out_shape.AddDim(batch_size); in BatchExampleProtoToTensors()
349 for (const int dim : shape.dim_sizes()) out_shape.AddDim(dim); in BatchExampleProtoToTensors()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dreshape_op.cc62 shape.AddDim(1); in Compile()
67 shape.AddDim(size); in Compile()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_resource.cc137 ta_shape.AddDim(max_array_size_); in SetZeroValue()
145 ta_shape.AddDim(max_array_size_); in SetZeroValue()
170 ta_shape.AddDim(max_array_size_); in GetOrCreateTensorArrayGradient()
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dmodel_ops.cc196 output_shape.AddDim(num_data); in Compute()
197 output_shape.AddDim(num_outputs); in Compute()
224 output_paths_shape.AddDim(param_proto_.inference_tree_paths() ? num_data in Compute()
296 output_shape.AddDim(num_data); in Compute()
397 output_shape.AddDim(param_proto_.num_features()); in Compute()

12345