Searched refs:input_r4 (Results 1 – 2 of 2) sorted by relevance
340 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local352 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()400 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local416 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()466 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local484 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()536 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local549 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()600 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in RunTest() local615 client_->TransferToServer(input_r4).ConsumeValueOrDie(); in RunTest()[all …]
1476 auto input_r4 = input_r1.Reshape(input_dims).ConsumeValueOrDie(); in TEST_P() local1478 b.AddInstruction(HloInstruction::CreateConstant(std::move(input_r4))); in TEST_P()