Searched refs:RemoveDimRange (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape_test.cc | 86 TEST(TensorShapeTest, RemoveDimRange) { in TEST() argument
90 s0.RemoveDimRange(i, i); in TEST()
96 s0.RemoveDimRange(3, 1); // Empty interval. in TEST()
99 s0.RemoveDimRange(0, 3); in TEST()
103 s1.RemoveDimRange(2, 3); in TEST()
109 s2.RemoveDimRange(-2, -3); // Empty interval. in TEST()
112 s2.RemoveDimRange(0, -2); in TEST()
116 s3.RemoveDimRange(-3, -2); in TEST()
|
| D | tensor_shape.h | 243 RemoveDimRange(d, d + 1); in RemoveDim()
260 RemoveDimRange(dims() - n, dims()); in RemoveLastDims()
278 void RemoveDimRange(int begin, int end);
|
| D | tensor_shape.cc | 631 void TensorShapeBase<Shape>::RemoveDimRange(int begin, int end) { in RemoveDimRange() function in tensorflow::TensorShapeBase
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 70 input_shape_pre_axis.RemoveDimRange(axis, input_shape.dims()); in XlaGather()
72 input_shape_post_axis.RemoveDimRange(0, axis + num_index_dims); in XlaGather()
|
| D | segment_reduction_ops.cc | 82 buffer_shape.RemoveDimRange(0, indices_shape.dims()); in Compile()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | segment_reduction_ops.cc | 72 buffer_shape.RemoveDimRange(0, indices_shape.dims()); in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_tensor_op.cc | 454 element_shape.RemoveDimRange(0, ragged_rank + 1); in SetOutput()
|