Searched defs:arg_buffers (Results 1 – 6 of 6) sorted by relevance
95 std::vector<PjRtBuffer*> arg_buffers(args.size()); in Execute() local117 std::vector<std::vector<PjRtBuffer*>> arg_buffers(args.size()); in ExecuteOnLocalDevices() local159 std::vector<std::vector<PjRtBuffer*>> arg_buffers(num_computations); in ExecuteShardedOnLocalDevices() local
126 std::vector<xla::PjRtBuffer*> arg_buffers; member
315 std::vector<std::vector<xla::PyBuffer*>> arg_buffers; in Call() local
945 std::vector<xla::PjRtBuffer*>& arg_buffers = arguments.arg_buffers; in ConvertArgsToBuffers() local
110 auto* arg_buffers = arg.MutableBuffers(); in ExecuteAsyncOnStream() local
773 std::vector<std::vector<PyTpuBuffer*>> arg_buffers(num_computations); in ExecuteShardedOnLocalDevices() local