Searched refs:argument_count_provider (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.h | 131 std::function<int64(int64)> argument_count_provider,
158 std::function<int64(int64)> argument_count_provider,
|
| D | hlo_runner.cc | 197 std::function<int64(int64)> argument_count_provider, in ExecuteReplicatedImpl() argument
211 total += argument_count_provider(i); in ExecuteReplicatedImpl()
234 const int64 argument_count = argument_count_provider(i); in ExecuteReplicatedImpl()
370 std::function<int64(int64)> argument_count_provider, in ExecuteReplicated() argument
414 argument_count_provider, argument_provider, options, &device_assignment); in ExecuteReplicated()
|
| D | hlo_runner_interface.h | 169 std::function<int64(int64)> argument_count_provider,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 175 std::function<int64(int64)> argument_count_provider,
|
| D | hlo_test_base.cc | 235 std::function<int64(int64)> argument_count_provider, in ExecuteReplicated() argument
243 executable_provider, argument_count_provider, argument_provider, options); in ExecuteReplicated()
|