| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | lower_upper_bound_ops.cc | 50 new_values_shape.InsertDim(/* d */ 2, /* size */ 1); in BuildLowerUpperBoundOp()
56 new_sorted_inputs_shape.InsertDim(/* d */ 1, /* size */ 1); in BuildLowerUpperBoundOp()
|
| D | pack_op.cc | 75 child_shape.InsertDim(axis, 1); in Compile()
|
| D | segment_reduction_ops.cc | 83 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/ |
| D | pad_op.cc | 163 collapsed_input_shape->InsertDim(collapsed_input_shape->dims(), in CollapseAdjacentNonPaddedDimensions()
165 collapsed_output_shape->InsertDim(collapsed_output_shape->dims(), in CollapseAdjacentNonPaddedDimensions()
182 collapsed_input_shape->InsertDim(collapsed_input_shape->dims(), in CollapseAdjacentNonPaddedDimensions()
184 collapsed_output_shape->InsertDim(collapsed_output_shape->dims(), in CollapseAdjacentNonPaddedDimensions()
|
| D | pack_op.cc | 63 output_shape.InsertDim(axis, num); in Compute()
|
| D | one_hot_op.cc | 91 output_shape.InsertDim(axis, depth_v); in Compute()
|
| D | map_kernels.h | 199 output_shape.InsertDim(0, map->tensors().size()); in Compute()
|
| D | list_kernels.h | 115 output_shape.InsertDim(0, tensor_list->tensors().size()); in Compute()
417 output_shape.InsertDim(0, leading_dim); in Compute()
596 output_shape.InsertDim(0, indices.NumElements()); in Compute()
|
| D | tensor_array_ops.cc | 647 empty_shape.InsertDim(0, 0); in Compute()
668 output_shape.InsertDim(0, num_indices); in Compute()
851 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 | 274 void InsertDim(int d, int64 size);
440 void TensorShapeOld::InsertDim(int d, int64 size) { in InsertDim() function in tensorflow::__anon127dd62f0111::TensorShapeOld
580 s.InsertDim(dim, sz); in TEST()
581 sold.InsertDim(dim, sz); in TEST()
|
| D | tensor_shape.h | 222 void InsertDim(int d, int64 size);
|
| D | tensor_shape.cc | 506 void TensorShapeBase<Shape>::InsertDim(int d, int64 size) { in InsertDim() function in tensorflow::TensorShapeBase
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | segment_reduction_ops.cc | 73 buffer_shape.InsertDim(0, num_segments); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | map_defun_op.cc | 149 actual_shape.InsertDim(0, compute_opts_->batch_size); in SetRetval()
300 output_shape.InsertDim(0, opts->batch_size); in SetupOutputs()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | utils.h | 166 shape->InsertDim(0, batch_size.value());
|
| D | utils.cc | 201 shape->InsertDim(0, batch_size.value()); in TrtDimsToTensorShape()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 93 output_shape.InsertDim(axis, depth); in OneHot()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | base_binary_ops_test.h | 218 expected_shape.InsertDim(0, 1); in TestOneEffectiveScalar()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 2501 total_shape.InsertDim(0, -1); in ParseSequenceDenseFeatures()
|