Searched refs:argument_layout (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_metadata_test.cc | 49 Shape argument_layout = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
53 {&argument_layout, &argument_layout}, in TEST_F()
73 Shape argument_layout = ShapeUtil::MakeShape(F32, {}); in TEST_F() local
77 {&argument_layout, &argument_layout}, in TEST_F()
|
| D | local_client_execute_test.cc | 761 Shape argument_layout = in XLA_TEST_F() local
765 local_client_->Compile(builder.Build().ValueOrDie(), {&argument_layout}, in XLA_TEST_F()
798 Shape argument_layout = in XLA_TEST_F() local
804 local_client_->Compile(builder.Build().ValueOrDie(), {&argument_layout}, in XLA_TEST_F()
820 Shape argument_layout = in XLA_TEST_F() local
824 local_client_->Compile(builder.Build().ValueOrDie(), {&argument_layout}, in XLA_TEST_F()
|