Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.cc266 int num_output_buffers = output_buffers[0].size(); in ExecuteShardedOnLocalDevicesInternal() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc442 int64_t num_output_buffers = ShapeUtil::SubshapeCount(instr1.shape()) + in FusionFitsInBudget() local
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc791 int num_output_buffers = output_buffers[0].size(); in ExecuteShardedOnLocalDevices() local