Searched refs:input_r4 (Results 1 – 2 of 2) sorted by relevance
508 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local520 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()568 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local584 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()634 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local652 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()704 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local717 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()768 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local783 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()[all …]
1399 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in TEST_P() local1401 b.AddInstruction(HloInstruction::CreateConstant(std::move(input_r4))); in TEST_P()