Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.cc221 std::vector<absl::Span<const ShapedBuffer* const>> argument_buffer_slices; in ExecuteReplicatedImpl() local
247 argument_buffer_slices.emplace_back( in ExecuteReplicatedImpl()
305 execution_helper(service_run_options, argument_buffer_slices)); in ExecuteReplicatedImpl()
325 argument_buffer_slices) in ExecuteReplicated()
331 argument_buffer_slices)); in ExecuteReplicated()
344 &service_run_options[i], argument_buffer_slices[i], in ExecuteReplicated()
379 argument_buffer_slices) in ExecuteReplicated()
392 for (const auto& arg : argument_buffer_slices[i]) { in ExecuteReplicated()
397 &service_run_options[i], argument_buffer_slices[i], nullptr); in ExecuteReplicated()