Searched refs:ReshapeSlice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reshape_test.cc | 737 Literal expected = LiteralUtil::ReshapeSlice({2, 1}, {1, 0}, input_literal); in XLA_TEST_P()
757 Literal expected = LiteralUtil::ReshapeSlice({4, 2}, {1, 0}, input_literal); in XLA_TEST_P()
891 LiteralUtil::ReshapeSlice(new_bounds, {2, 3, 1, 0}, input_literal) in XLA_TEST_P()
919 LiteralUtil::ReshapeSlice(new_bounds, {2, 3, 1, 0}, input_literal) in XLA_TEST_P()
947 LiteralUtil::ReshapeSlice(new_bounds, {2, 3, 1, 0}, input_literal) in XLA_TEST_P()
976 LiteralUtil::ReshapeSlice(new_bounds, {2, 3, 1, 0}, input_literal) in XLA_TEST_P()
1004 LiteralUtil::ReshapeSlice(new_bounds, {1, 0, 2, 3}, input_literal) in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 250 static Literal ReshapeSlice(absl::Span<const int64> new_dimensions,
|
| D | literal_util.cc | 344 /* static */ Literal LiteralUtil::ReshapeSlice( in ReshapeSlice() function in xla::LiteralUtil
|