/external/tensorflow/tensorflow/core/kernels/ |
D | batchtospace_op.cc | 137 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()
|
D | spacetobatch_op.cc | 134 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()
|
D | svd_op_gpu.cu.cc | 113 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 …]
|
D | reduction_ops_common.cc | 24 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()
|
D | matrix_diag_op.cc | 63 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()
|
D | reshape_op.h | 109 shape->AddDim(1); in ValidateSizes() 114 shape->AddDim(size); in ValidateSizes()
|
D | diag_op.cc | 55 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()
|
D | decode_raw_op.cc | 54 out_shape.AddDim(0); in Compute() 66 out_shape.AddDim(added_dim); in Compute()
|
D | mkl_batch_matmul_op.cc | 75 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/ |
D | k_feature_gradient_op.cc | 113 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()
|
D | stochastic_hard_routing_gradient_op.cc | 134 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()
|
D | hard_routing_function_op.cc | 116 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()
|
D | stochastic_hard_routing_function_op.cc | 129 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()
|
D | unpack_path_op.cc | 86 output_shape.AddDim(num_data); in Compute() 87 output_shape.AddDim(num_nodes); in Compute()
|
D | routing_function_op.cc | 107 output_shape.AddDim(num_data); in Compute() 108 output_shape.AddDim(max_nodes_); in Compute()
|
D | routing_gradient_op.cc | 106 output_shape.AddDim(num_data); in Compute() 107 output_shape.AddDim(max_nodes_); in Compute()
|
D | k_feature_routing_function_op.cc | 122 output_shape.AddDim(num_data); in Compute() 123 output_shape.AddDim(max_nodes_); in Compute()
|
/external/tensorflow/tensorflow/core/framework/ |
D | tensor_shape_test.cc | 67 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 …]
|
D | tensor_shape.cc | 130 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/ |
D | strided_slice_op.cc | 266 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()
|
D | example_proto_fast_parsing.cc | 1002 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 …]
|
D | example_proto_helper.cc | 310 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/ |
D | reshape_op.cc | 62 shape.AddDim(1); in Compile() 67 shape.AddDim(size); in Compile()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_resource.cc | 137 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/ |
D | model_ops.cc | 196 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()
|