Home
last modified time | relevance | path

Searched defs:arg_buffers (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_executable.cc129 std::vector<PjRtBuffer*> arg_buffers(args.size()); in ExecuteInternal() local
242 std::vector<std::vector<PjRtBuffer*>> arg_buffers(num_computations); in ExecuteShardedOnLocalDevicesInternal() local
H A Djax_jit.h146 std::vector<xla::PjRtBuffer*> arg_buffers; member
H A Djax_jit.cc740 std::vector<xla::PjRtBuffer*>& arg_buffers = arguments.arg_buffers; in CopyBuffersToDevice() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executable.cc86 auto* arg_buffers = arg.MutableBuffers(); in ExecuteAsyncOnStream() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client.cc783 std::vector<std::vector<PyTpuBuffer*>> arg_buffers(num_computations); in ExecuteShardedOnLocalDevices() local