Searched refs:CopyLayoutBetweenShapes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util_test.cc | 77 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
83 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
90 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
100 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
106 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
113 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
133 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
152 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
159 ASSERT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
166 ASSERT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
[all …]
|
| D | layout_util.h | 179 static Status CopyLayoutBetweenShapes(const Shape& src, Shape* dst);
|
| D | layout_util.cc | 409 Status LayoutUtil::CopyLayoutBetweenShapes(const Shape& src, Shape* dst) { in CopyLayoutBetweenShapes() function in xla::LayoutUtil
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 808 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in CreateCopyWithNewLayout()
818 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in CreateCopyWithNewLayout()
1548 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1555 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1560 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1566 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
|
| D | algebraic_simplifier.cc | 2269 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in HandlePad()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 166 LayoutUtil::CopyLayoutBetweenShapes(fusion->shape(), &update_shape)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
|