Searched refs:ReshapeSlice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reshape_test.cc | 701 Literal expected = LiteralUtil::ReshapeSlice({2, 1}, {1, 0}, input_literal); in XLA_TEST_P()
720 Literal expected = LiteralUtil::ReshapeSlice({4, 2}, {1, 0}, input_literal); in XLA_TEST_P()
849 LiteralUtil::ReshapeSlice(new_bounds, {2, 3, 1, 0}, input_literal) in XLA_TEST_P()
876 LiteralUtil::ReshapeSlice(new_bounds, {2, 3, 1, 0}, input_literal) in XLA_TEST_P()
903 LiteralUtil::ReshapeSlice(new_bounds, {2, 3, 1, 0}, input_literal) in XLA_TEST_P()
931 LiteralUtil::ReshapeSlice(new_bounds, {2, 3, 1, 0}, input_literal) in XLA_TEST_P()
958 LiteralUtil::ReshapeSlice(new_bounds, {1, 0, 2, 3}, input_literal) in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.cc | 292 /* static */ Literal LiteralUtil::ReshapeSlice( in ReshapeSlice() function in xla::LiteralUtil
|
| D | literal_util.h | 268 static Literal ReshapeSlice(absl::Span<const int64> new_dimensions,
|