Home
last modified time | relevance | path

Searched refs:FindTranspose021 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc819 EXPECT_EQ(std::nullopt, ShapeUtil::FindTranspose021(shape, transposed)); in TEST()
822 TEST(ShapeUtilTest, FindTranspose021) { in TEST() argument
826 ShapeUtil::FindTranspose021(shape, transposed)); in TEST()
834 ShapeUtil::FindTranspose021(shape, transposed)); in TEST()
843 ShapeUtil::FindTranspose021(shape, transposed)); in TEST()
Dshape_util.h734 static std::optional<Vector3> FindTranspose021(const Shape& a,
Dshape_util.cc1692 /* static */ std::optional<Vector3> ShapeUtil::FindTranspose021( in FindTranspose021() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emission_utils.cc790 ShapeUtil::FindTranspose021(operand_shapes[operand_idx], output_shape); in FindTranspose021DimsAndParameters()