Searched defs:arg_literal (Results 1 – 6 of 6) sorted by relevance
212 Literal arg_literal = LiteralUtil::CreateR1<int64_t>({arg}); in XLA_TEST_F() local231 Literal arg_literal = LiteralUtil::CreateR1<uint32_t>({arg}); in XLA_TEST_F() local249 Literal arg_literal = LiteralUtil::CreateR1<float>({arg}); in XLA_TEST_F() local266 Literal arg_literal = LiteralUtil::CreateR1<uint32_t>({arg}); in XLA_TEST_F() local283 Literal arg_literal = LiteralUtil::CreateR1<int32_t>({arg}); in XLA_TEST_F() local320 Literal arg_literal = LiteralUtil::CreateR1<float>({arg}); in XLA_TEST_F() local
350 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local382 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local438 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F() local495 auto arg_literal = in XLA_TEST_F() local535 Literal arg_literal = LiteralUtil::CreateR0<float>(123.0); in XLA_TEST_F() local
384 Literal arg_literal(shape); in XLA_TEST_P() local404 Literal arg_literal = in XLA_TEST_P() local
184 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in ReduceWindowMaxIotaTest() local2634 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local2676 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local2793 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local2845 Literal arg_literal = in TEST_P() local
3038 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleAsyncStart() local3128 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleCall() local3164 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleFusion() local
2222 auto arg_literal = LiteralUtil::CreateR1(b); in Propagate() local2339 auto arg_literal = LiteralUtil::CreateR1(b); in SelectValidPortion() local