Searched refs:input_r1 (Results 1 – 3 of 3) sorted by relevance
276 auto input_r1 = LiteralUtil::CreateR1<float>(input_elems); in XLA_TEST_F() local277 auto input_r5 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in XLA_TEST_F()339 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest() local340 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()399 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest() local400 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()465 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest() local466 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()535 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest() local536 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()[all …]
109 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in TestImpl() local110 auto input_r3 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in TestImpl()
1475 auto input_r1 = LiteralUtil::CreateR1<float>(input_elems); in TEST_P() local1476 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in TEST_P()