Home
last modified time | relevance | path

Searched refs:LayoutsInShapesEqual (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dlayout_util_test.cc56 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple0)); in TEST_F()
57 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple1)); in TEST_F()
58 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple0, tuple2)); in TEST_F()
59 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple1, tuple0)); in TEST_F()
60 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple2, tuple0)); in TEST_F()
62 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple1, tuple1)); in TEST_F()
63 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple1, tuple2)); in TEST_F()
64 EXPECT_FALSE(LayoutUtil::LayoutsInShapesEqual(tuple2, tuple1)); in TEST_F()
67 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple2, tuple2)); in TEST_F()
68 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(tuple2, other_tuple2)); in TEST_F()
[all …]
Dlayout_util.h187 static bool LayoutsInShapesEqual(const Shape& lhs, const Shape& rhs);
Dlayout_util.cc413 /* static */ bool LayoutUtil::LayoutsInShapesEqual(const Shape& lhs, in LayoutsInShapesEqual() function in xla::LayoutUtil
421 if (!LayoutsInShapesEqual(lhs.tuple_shapes(i), rhs.tuple_shapes(i))) { in LayoutsInShapesEqual()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_thunk.cc97 LayoutUtil::LayoutsInShapesEqual(hlo->shape(), hlo->operand(0)->shape())); in CudnnBatchNormForwardInferenceThunk()
156 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0), in CudnnBatchNormForwardTrainingThunk()
238 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0), in CudnnBatchNormBackwardThunk()
240 CHECK(LayoutUtil::LayoutsInShapesEqual(hlo->shape().tuple_shapes(0), in CudnnBatchNormBackwardThunk()
Dir_emitter_unnested.cc1218 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual(keys_shape, in HandleSort()
1220 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual( in HandleSort()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc240 LayoutUtil::LayoutsInShapesEqual(constant0->shape(), constant1->shape())); in TEST_F()
243 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual( in TEST_F()
245 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual( in TEST_F()
281 EXPECT_TRUE(LayoutUtil::LayoutsInShapesEqual(result_shape, select->shape())); in TEST_F()
Dlayout_assignment.cc677 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual( in CheckCustomCallLayout()
703 LayoutUtil::LayoutsInShapesEqual(init_shape, while_inst->shape())); in CheckWhileLayout()
733 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual( in CheckFusionLayout()
736 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual( in CheckFusionLayout()
760 if (!LayoutUtil::LayoutsInShapesEqual(constant->literal().shape(), in CheckConstantLayout()
Dservice.cc962 if (LayoutUtil::LayoutsInShapesEqual(return_shape, result_literal.shape())) { in TransferToClient()
Dhlo_evaluator_test.cc525 ASSERT_TRUE(LayoutUtil::LayoutsInShapesEqual(input_literal.shape(), in TEST_P()
545 ASSERT_FALSE(LayoutUtil::LayoutsInShapesEqual(input_literal.shape(), in TEST_P()
Dhlo_evaluator.cc1607 if (!LayoutUtil::LayoutsInShapesEqual(GetEvaluatedLiteralFor(hlo).shape(), in Postprocess()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.cc537 LayoutUtil::LayoutsInShapesEqual(keys_shape, operand->shape())); in HandleSort()
538 TF_RET_CHECK(LayoutUtil::LayoutsInShapesEqual( in HandleSort()