Searched refs:MakeFakeArguments (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils_test.cc | 56 TF_ASSERT_OK(MakeFakeArguments(&module).status()); in XLA_TEST_F()
74 MakeFakeArguments(module.get())); in XLA_TEST_F()
105 MakeFakeArguments(module.get())); in XLA_TEST_F()
139 MakeFakeArguments(module.get())); in XLA_TEST_F()
170 MakeFakeArguments(module.get())); in XLA_TEST_F()
201 MakeFakeArguments(module.get())); in XLA_TEST_F()
230 MakeFakeArguments(module.get())); in XLA_TEST_F()
259 MakeFakeArguments(module.get())); in XLA_TEST_F()
297 MakeFakeArguments(module.get())); in XLA_TEST_F()
|
| D | test_utils.h | 93 StatusOr<std::vector<Literal>> MakeFakeArguments(HloModule* const module,
100 StatusOr<std::vector<Literal>> MakeFakeArguments(HloModule* const module,
|
| D | hlo_test_base.cc | 315 auto fake_arguments = MakeFakeArguments(module.get()).ConsumeValueOrDie(); in RunAndCompare()
330 MakeFakeArguments(module.get()).ConsumeValueOrDie(); in RunAndCompareNoHloPasses()
343 MakeFakeArguments(module.get()).ConsumeValueOrDie(); in Run()
382 MakeFakeArguments(module.get()).ConsumeValueOrDie(); in Run()
431 MakeFakeArguments(module.get()).ConsumeValueOrDie(); in RunReplicated()
477 fake_arguments[i] = MakeFakeArguments(module.get()).ConsumeValueOrDie(); in RunMultipleTimes()
|
| D | test_utils.cc | 673 StatusOr<std::vector<Literal>> MakeFakeArguments(HloModule* const module, in MakeFakeArguments() function
678 return MakeFakeArguments(module, engine.get(), use_large_range); in MakeFakeArguments()
681 StatusOr<std::vector<Literal>> MakeFakeArguments(HloModule* const module, in MakeFakeArguments() function
|
| D | grouped_convolution_test.cc | 270 TF_ASSERT_OK_AND_ASSIGN(auto fake_arguments, MakeFakeArguments(module.get())); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_update_slice_test.cc | 70 TF_ASSERT_OK_AND_ASSIGN(auto fake_arguments, MakeFakeArguments(module.get())); in XLA_TEST_F()
|
| D | hlo_evaluator_test.cc | 4216 auto args = MakeFakeArguments(m_.get()).ConsumeValueOrDie(); in TEST_P()
4355 auto args = MakeFakeArguments(m_.get()).ConsumeValueOrDie(); in TEST_F()
4377 auto args = MakeFakeArguments(m_.get()).ConsumeValueOrDie(); in TEST_F()
4399 auto args = MakeFakeArguments(m_.get()).ConsumeValueOrDie(); in TEST_F()
4418 auto args = MakeFakeArguments(m_.get()).ConsumeValueOrDie(); in TEST_F()
4435 auto args = MakeFakeArguments(m_.get()).ConsumeValueOrDie(); in TEST_F()
4460 auto args = MakeFakeArguments(m_.get()).ConsumeValueOrDie(); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | run_hlo_module.cc | 141 std::vector<Literal> args = MakeFakeArguments(test_module.get(), engine, in RunAndCompare()
|