Searched defs:arg_literal (Results 1 – 6 of 6) sorted by relevance
213 Literal arg_literal = LiteralUtil::CreateR1<int64>({arg}); in XLA_TEST_F() local232 Literal arg_literal = LiteralUtil::CreateR1<uint32>({arg}); in XLA_TEST_F() local250 Literal arg_literal = LiteralUtil::CreateR1<float>({arg}); in XLA_TEST_F() local267 Literal arg_literal = LiteralUtil::CreateR1<uint32>({arg}); in XLA_TEST_F() local284 Literal arg_literal = LiteralUtil::CreateR1<int32>({arg}); in XLA_TEST_F() local321 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() local436 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F() local489 auto arg_literal = in XLA_TEST_F() local529 Literal arg_literal = LiteralUtil::CreateR0<float>(123.0); in XLA_TEST_F() local
386 Literal arg_literal(shape); in XLA_TEST_P() local406 Literal arg_literal = in XLA_TEST_P() local
159 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in ReduceWindowMaxIotaTest() local2596 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local2638 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local2755 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local2807 Literal arg_literal = in TEST_P() local
1585 auto arg_literal = LiteralUtil::CreateR1(b); in Propagate() local1679 auto arg_literal = LiteralUtil::CreateR1(b); in SelectValidPortion() local
1924 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleCall() local1959 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleFusion() local