Home
last modified time | relevance | path

Searched refs:filter_r4 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvolution_test.cc345 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local
354 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
405 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local
418 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
471 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local
486 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
541 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local
551 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
605 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in RunTest() local
617 client_->TransferToServer(filter_r4).ConsumeValueOrDie(); in RunTest()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc1483 auto filter_r4 = filter_r1.Reshape(filter_dims).ConsumeValueOrDie(); in TEST_P() local
1485 b.AddInstruction(HloInstruction::CreateConstant(std::move(filter_r4))); in TEST_P()