| /external/tensorflow/tensorflow/core/kernels/ |
| D | pad_op.cc | 172 collapsed_input_shape->InsertDim(collapsed_input_shape->dims(), in CollapseAdjacentNonPaddedDimensions()
174 collapsed_output_shape->InsertDim(collapsed_output_shape->dims(), in CollapseAdjacentNonPaddedDimensions()
191 collapsed_input_shape->InsertDim(collapsed_input_shape->dims(), in CollapseAdjacentNonPaddedDimensions()
193 collapsed_output_shape->InsertDim(collapsed_output_shape->dims(), in CollapseAdjacentNonPaddedDimensions()
|
| D | pack_op.cc | 76 output_shape.InsertDim(axis, num); in Compute()
|
| D | one_hot_op.cc | 90 output_shape.InsertDim(axis, depth_v); in Compute()
|
| D | list_kernels.h | 136 output_shape.InsertDim(0, tensor_list->tensors.size()); in Compute()
442 output_shape.InsertDim(0, leading_dim); in Compute()
619 output_shape.InsertDim(0, indices.NumElements()); in Compute()
|
| D | tensor_array_ops.cc | 652 empty_shape.InsertDim(0, 0); in Compute()
673 output_shape.InsertDim(0, num_indices); in Compute()
857 empty_shape.InsertDim(0, 0); in Compute()
|
| D | barrier_ops.cc | 163 component_shape.InsertDim(0, insertion_size); in TryInsertMany()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape_test.cc | 251 void InsertDim(int d, int64 size);
417 void TensorShapeOld::InsertDim(int d, int64 size) { in InsertDim() function in tensorflow::__anone850e9110111::TensorShapeOld
557 s.InsertDim(dim, sz); in TEST()
558 sold.InsertDim(dim, sz); in TEST()
|
| D | tensor_shape.h | 197 void InsertDim(int d, int64 size);
|
| D | tensor_shape.cc | 388 void TensorShapeBase<Shape>::InsertDim(int d, int64 size) { in InsertDim() function in tensorflow::TensorShapeBase
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | tensor_list_ops.cc | 333 output_list_shape.InsertDim(0, leading_dim); in GetInitializedList()
367 slice_shape.InsertDim(0, 1LL); in Compile()
410 slice_shape.InsertDim(0, 1LL); in Compile()
|
| D | pack_op.cc | 75 child_shape.InsertDim(axis, 1); in Compile()
|
| D | segment_reduction_ops.cc | 79 buffer_shape.InsertDim(0, num_segments); in Compile()
|
| D | stack_ops.cc | 153 slice_shape.InsertDim(0, 1LL); in Compile()
|
| D | tensor_array_ops.cc | 116 shape->InsertDim(0, resource->max_array_size()); in GetTensorArrayShape()
220 slice_shape.InsertDim(0, 1LL); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 234 actual_shape.InsertDim(0, compute_opts_->batch_size); in SetRetval()
310 output_shape.InsertDim(0, opts->batch_size); in SetupOutputs()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 95 output_shape.InsertDim(axis, depth); in OneHot()
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/kernels/ |
| D | stats_accumulator_ops.cc | 191 gradient_shape.InsertDim(0, num_slots); in SerializeTensorAccumulatorToOutput()
200 hessian_shape.InsertDim(0, num_slots); in SerializeTensorAccumulatorToOutput()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 1200 attrs.shape_.InsertDim(0, batch_size); in AddTestTensor()
|