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()
|