Searched refs:filter_r4 (Results 1 – 2 of 2) sorted by relevance
345 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local354 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()405 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local418 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()471 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local486 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()541 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local551 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()605 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local617 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()[all …]
1483 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in TEST_P() local1485 b.AddInstruction(HloInstruction::CreateConstant(std::move(filter_r4))); in TEST_P()