Searched refs:CopyLayoutBetweenShapes (Results 1 – 7 of 7) 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 | 169 static Status CopyLayoutBetweenShapes(const Shape& src, Shape* dst);
|
| D | layout_util.cc | 399 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 | 216 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in EmitFusedDynamicUpdateSliceInPlaceImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment.cc | 836 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in CreateCopyWithNewLayout()
846 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in CreateCopyWithNewLayout()
1689 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1696 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1701 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1707 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
|
| D | algebraic_simplifier.cc | 3481 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in HandlePad()
3509 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in HandlePad()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 2913 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes(hlo->outfeed_shape(), in HandleOutfeed()
3026 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in HandleOutfeed()
3703 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in Run()
3707 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in Run()
|