Searched refs:argument_provider (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.h | 132 std::function<const Literal*(int64, int64)> argument_provider,
159 std::function<const Literal*(int64, int64)> argument_provider,
|
| D | hlo_runner.cc | 198 std::function<const Literal*(int64, int64)> argument_provider, in ExecuteReplicatedImpl() argument
236 const Literal* const argument = argument_provider(i, arg_index); in ExecuteReplicatedImpl()
371 std::function<const Literal*(int64, int64)> argument_provider, in ExecuteReplicated() argument
414 argument_count_provider, argument_provider, options, &device_assignment); in ExecuteReplicated()
|
| D | hlo_runner_interface.h | 170 std::function<const Literal*(int64, int64)> argument_provider,
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | hlo_test_base.h | 176 std::function<const Literal*(int64, int64)> argument_provider,
|
| D | hlo_test_base.cc | 236 std::function<const Literal*(int64, int64)> argument_provider, in ExecuteReplicated() argument
243 executable_provider, argument_count_provider, argument_provider, options); in ExecuteReplicated()
|