Home
last modified time | relevance | path

Searched refs:CopyLayoutBetweenShapes (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc77 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 …]
Dlayout_util.h182 static tensorflow::Status CopyLayoutBetweenShapes(const Shape& src,
Dlayout_util.cc419 tensorflow::Status LayoutUtil::CopyLayoutBetweenShapes(const Shape& src, in CopyLayoutBetweenShapes() function in xla::LayoutUtil
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc92 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in CreateCopyWithNewLayout()
100 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in CreateCopyWithNewLayout()
1322 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1329 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1334 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
1340 TF_RETURN_IF_ERROR(LayoutUtil::CopyLayoutBetweenShapes( in SetFusionLayouts()
Dalgebraic_simplifier.cc1300 LayoutUtil::CopyLayoutBetweenShapes(pad->shape(), &nonzero_pad_shape)); in HandlePad()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dops.cc145 LayoutUtil::CopyLayoutBetweenShapes(fusion->shape(), &update_shape)); in EmitFusedDynamicUpdateSliceInPlaceImpl()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc531 return LayoutUtil::CopyLayoutBetweenShapes(operand->shape(), in HandleFusion()