Home
last modified time | relevance | path

Searched refs:RemoveDimRange (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dtensor_shape_test.cc85 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()
Dtensor_shape.h208 RemoveDimRange(d, d + 1); in RemoveDim()
215 RemoveDimRange(dims() - n, dims()); in RemoveLastDims()
222 void RemoveDimRange(int begin, int end);
Dtensor_shape.cc438 void TensorShapeBase<Shape>::RemoveDimRange(int begin, int end) { in RemoveDimRange() function in tensorflow::TensorShapeBase
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc64 input_shape_pre_axis.RemoveDimRange(axis, input_shape.dims()); in XlaGather()
66 input_shape_post_axis.RemoveDimRange(0, axis + num_index_dims); in XlaGather()
Dsegment_reduction_ops.cc78 buffer_shape.RemoveDimRange(0, indices_shape.dims()); in Compile()