Searched defs:arg_buffers (Results 1 – 6 of 6) sorted by relevance
75 std::vector<PjRtBuffer*> arg_buffers(args.size()); in Execute() local111 std::vector<std::vector<PjRtBuffer*>> arg_buffers(num_computations); in ExecuteShardedOnLocalDevices() local
140 std::vector<xla::PjRtBuffer*> arg_buffers; member
254 std::vector<std::vector<xla::PyBuffer::object>> arg_buffers; in Call() local
648 std::vector<xla::PjRtBuffer*>& arg_buffers = arguments.arg_buffers; in CopyBuffersToDevice() local
85 auto* arg_buffers = arg.MutableBuffers(); in ExecuteAsyncOnStream() local
773 std::vector<std::vector<PyTpuBuffer*>> arg_buffers(num_computations); in ExecuteShardedOnLocalDevices() local