Searched refs:partition_buffers (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | parallel_cpu_executable.cc | 127 // If 'partition_buffers' is non-null, parallel task will be invoked on
129 // 'partition_buffers'.
130 void Schedule(HloInstruction* instruction, int64* partition_buffers,
137 // Returns in 'partition_buffers' the partition [size, limit) for each
211 auto partition_buffers = local
213 Schedule(instruction, partition_buffers, result_buffer);
257 void Executor::Schedule(HloInstruction* instruction, int64* partition_buffers, argument
265 partition_buffers, exec_run_options, function]() {
267 partition_buffers, profile_counters_array_);
270 delete[] partition_buffers;
293 auto partition_buffers = new int64[partition.size() * 2]; local
[all...] |
Completed in 134 milliseconds