Searched defs:arg_literal (Results 1 – 5 of 5) 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
347 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local379 auto arg_literal = LiteralUtil::MakeTupleFromSlices( in XLA_TEST_F() local433 Literal arg_literal = LiteralUtil::MakeTupleOwned(std::move(arg_elements)); in XLA_TEST_F() local486 auto arg_literal = in XLA_TEST_F() local526 Literal arg_literal = LiteralUtil::CreateR0<float>(123.0); in XLA_TEST_F() local
362 Literal arg_literal(shape); in XLA_TEST_P() local382 Literal arg_literal = in XLA_TEST_P() local
1510 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local1552 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local1604 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local1655 auto arg_literal = LiteralUtil::CreateR2FromArray2D<float>(*arg_array); in TEST_P() local1707 Literal arg_literal = in TEST_P() local
1212 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleCall() local1247 const Literal& arg_literal = GetEvaluatedLiteralFor(operand); in HandleFusion() local