Searched refs:argument_buffers (Results 1 – 3 of 3) sorted by relevance
51 std::vector<ShapedBuffer> argument_buffers; in ExecuteAsyncOnStream() local52 argument_buffers.reserve(arguments.size()); in ExecuteAsyncOnStream()59 argument_buffers.push_back(ShapedBuffer(buffers.shape(), in ExecuteAsyncOnStream()62 auto out_it = argument_buffers.back().buffers().begin(); in ExecuteAsyncOnStream()70 for (const auto& a : argument_buffers) { in ExecuteAsyncOnStream()86 const auto& actual_shape = argument_buffers[i].on_device_shape(); in ExecuteAsyncOnStream()114 run_options->stream(), argument_buffers[p])); in ExecuteAsyncOnStream()
95 TF_ASSIGN_OR_RETURN(std::vector<ScopedShapedBuffer> argument_buffers, in Execute()100 /*arguments=*/argument_buffers, in Execute()109 TF_ASSIGN_OR_RETURN(std::vector<ScopedShapedBuffer> argument_buffers, in ExecuteWithExecutable()114 /*arguments=*/argument_buffers, in ExecuteWithExecutable()204 std::vector<ScopedShapedBuffer> argument_buffers; in ExecuteReplicatedImpl() local215 argument_buffers.reserve(total_argument_count); in ExecuteReplicatedImpl()244 argument_buffers.push_back(std::move(argument_buffer)); in ExecuteReplicatedImpl()245 argument_buffer_ptrs[index++] = &argument_buffers.back(); in ExecuteReplicatedImpl()
313 compOpt.argument_buffers = true;334 compOpt.argument_buffers = false;