Searched refs:CopyLayoutBetweenShapes (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util_test.cc | 69 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
75 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
82 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
102 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
109 ASSERT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
116 auto status = LayoutUtil::CopyLayoutBetweenShapes(src, &dst); in TEST_F()
135 auto status = LayoutUtil::CopyLayoutBetweenShapes(src, &dst); in TEST_F()
147 auto status = LayoutUtil::CopyLayoutBetweenShapes(src, &dst); in TEST_F()
162 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
173 EXPECT_IS_OK(LayoutUtil::CopyLayoutBetweenShapes(src, &dst)); in TEST_F()
[all …]
|
| D | layout_util.h | 165 static Status CopyLayoutBetweenShapes(const Shape& src, Shape* dst);
|
| D | layout_util.cc | 392 Status LayoutUtil::CopyLayoutBetweenShapes(const Shape& src, Shape* dst) { in CopyLayoutBetweenShapes() function in xla::LayoutUtil
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | dynamic_update_slice_util.cc | 220 LayoutUtil::CopyLayoutBetweenShapes(fusion->shape(), &update_shape)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 835 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in CreateCopyWithNewLayout()
845 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in CreateCopyWithNewLayout()
1666 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1673 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1678 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1684 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
|
| D | algebraic_simplifier.cc | 2752 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in HandlePad()
2778 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in HandlePad()
|