Searched refs:output_literal (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1189 Literal* output_literal) { in CopyDataToOutput() argument
1195 OutputType* output_data = output_literal->data<OutputType>().data(); in CopyDataToOutput()
1246 Literal* output_literal) { in CopyDataToOutput() argument
1251 output_literal); in CopyDataToOutput()
1255 output_strides, output_literal); in CopyDataToOutput()
1381 Literal output_literal = Literal::CreateFromShape(output_shape); in HandleFft() local
1409 const auto output_lengths = GetDimensionLengths(output_literal); in HandleFft()
1413 const auto output_strides = ComputeStrides(output_lengths, output_literal); in HandleFft()
1435 output_strides, &output_literal); in HandleFft()
1444 evaluated_[fft] = std::move(output_literal); in HandleFft()
|
| D | hlo_evaluator_test.cc | 521 auto output_literal = LiteralUtil::CreateR3<int32>( in TEST_F() local
526 output_literal.shape(), literal_instruction, {1, 2})); in TEST_F()
531 EXPECT_TRUE(LiteralTestUtil::Equal(result, output_literal)); in TEST_F()
537 auto output_literal = LiteralUtil::CreateR2<int32>( in TEST_F() local
544 output_literal.shape(), literal_instruction, in TEST_F()
550 EXPECT_TRUE(LiteralTestUtil::Equal(result, output_literal)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reshape_test.cc | 812 Literal output_literal = in XLA_TEST_P() local
822 EXPECT_EQ(expected.data<bfloat16>(), output_literal.data<bfloat16>()); in XLA_TEST_P()
824 EXPECT_EQ(input_literal.data<float>(), output_literal.data<float>()); in XLA_TEST_P()
|