Home
last modified time | relevance | path

Searched defs:arg_literal (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvert_test.cc213 Literal arg_literal = LiteralUtil::CreateR1<int64>({arg}); in XLA_TEST_F() local
232 Literal arg_literal = LiteralUtil::CreateR1<uint32>({arg}); in XLA_TEST_F() local
250 Literal arg_literal = LiteralUtil::CreateR1<float>({arg}); in XLA_TEST_F() local
267 Literal arg_literal = LiteralUtil::CreateR1<uint32>({arg}); in XLA_TEST_F() local
284 Literal arg_literal = LiteralUtil::CreateR1<int32>({arg}); in XLA_TEST_F() local
321 Literal arg_literal = LiteralUtil::CreateR1<float>({arg}); in XLA_TEST_F() local
Dlocal_client_execute_test.cc350 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local
382 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local
436 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F() local
489 auto arg_literal = in XLA_TEST_F() local
529 Literal arg_literal = LiteralUtil::CreateR0<float>(123.0); in XLA_TEST_F() local
Dreduce_window_test.cc386 Literal arg_literal(shape); in XLA_TEST_P() local
406 Literal arg_literal = in XLA_TEST_P() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc159 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in ReduceWindowMaxIotaTest() local
2596 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
2638 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
2755 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local
2807 Literal arg_literal = in TEST_P() local
Dspace_to_batch_converter.cc1585 auto arg_literal = LiteralUtil::CreateR1(b); in Propagate() local
1679 auto arg_literal = LiteralUtil::CreateR1(b); in SelectValidPortion() local
Dhlo_evaluator.cc1924 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleCall() local
1959 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleFusion() local