Home
last modified time | relevance | path

Searched refs:output_literal (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc773 Literal output_literal = in XLA_TEST_P() local
783 EXPECT_EQ(expected.data<bfloat16>(), output_literal.data<bfloat16>()); in XLA_TEST_P()
785 EXPECT_EQ(input_literal.data<float>(), output_literal.data<float>()); in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc444 auto output_literal = LiteralUtil::CreateR3<int32>( in TEST_F() local
449 output_literal.shape(), literal_instruction, {1, 2})); in TEST_F()
454 EXPECT_TRUE(LiteralTestUtil::Equal(result, output_literal)); in TEST_F()
460 auto output_literal = LiteralUtil::CreateR2<int32>( in TEST_F() local
467 output_literal.shape(), literal_instruction, in TEST_F()
473 EXPECT_TRUE(LiteralTestUtil::Equal(result, output_literal)); in TEST_F()