Searched refs:RemoveDimRange (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_shape_test.cc | 85 TEST(TensorShapeTest, RemoveDimRange) { in TEST() argument
89 s0.RemoveDimRange(i, i); in TEST()
95 s0.RemoveDimRange(3, 1); // Empty interval. in TEST()
98 s0.RemoveDimRange(0, 3); in TEST()
102 s1.RemoveDimRange(2, 3); in TEST()
108 s2.RemoveDimRange(-2, -3); // Empty interval. in TEST()
111 s2.RemoveDimRange(0, -2); in TEST()
115 s3.RemoveDimRange(-3, -2); in TEST()
|
| D | tensor_shape.h | 208 RemoveDimRange(d, d + 1); in RemoveDim()
215 RemoveDimRange(dims() - n, dims()); in RemoveLastDims()
222 void RemoveDimRange(int begin, int end);
|
| D | tensor_shape.cc | 438 void TensorShapeBase<Shape>::RemoveDimRange(int begin, int end) { in RemoveDimRange() function in tensorflow::TensorShapeBase
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 64 input_shape_pre_axis.RemoveDimRange(axis, input_shape.dims()); in XlaGather()
66 input_shape_post_axis.RemoveDimRange(0, axis + num_index_dims); in XlaGather()
|
| D | segment_reduction_ops.cc | 78 buffer_shape.RemoveDimRange(0, indices_shape.dims()); in Compile()
|