Searched refs:input_r1 (Results 1 – 2 of 2) sorted by relevance
444 auto input_r1 = LiteralUtil::CreateR1<float>(input_elems); in XLA_TEST_F() local445 auto input_r5 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in XLA_TEST_F()507 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest() local508 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()567 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest() local568 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()633 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest() local634 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()703 auto input_r1 = LiteralUtil::CreateR1<T>(input_elems); in RunTest() local704 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest()[all …]
1398 auto input_r1 = LiteralUtil::CreateR1<float>(input_elems); in TEST_P() local1399 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in TEST_P()