Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.cc171 int num_output_buffers = output_buffers[0].size(); in ExecuteShardedOnLocalDevices() local
173 outputs.resize(num_output_buffers); in ExecuteShardedOnLocalDevices()
174 for (int buffer_id = 0; buffer_id < num_output_buffers; ++buffer_id) { in ExecuteShardedOnLocalDevices()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc396 int64 num_output_buffers = ShapeUtil::SubshapeCount(instr1.shape()) + in FusionWouldBeTooLarge() local
407 num_output_buffers <= in FusionWouldBeTooLarge()
415 << " and num_output_buffers = " << num_output_buffers in FusionWouldBeTooLarge()
439 return operands.size() + num_output_buffers > kMaxOperandsAndOutputsPerFusion; in FusionWouldBeTooLarge()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc781 int num_output_buffers = output_buffers[0].size(); in ExecuteShardedOnLocalDevices() local
783 outputs.resize(num_output_buffers); in ExecuteShardedOnLocalDevices()
784 for (int buffer_id = 0; buffer_id < num_output_buffers; ++buffer_id) { in ExecuteShardedOnLocalDevices()