Searched refs:argument_buffers (Results 1 – 2 of 2) sorted by relevance
144 TF_ASSIGN_OR_RETURN(std::vector<ScopedShapedBuffer> argument_buffers, in Execute()149 /*arguments=*/argument_buffers, in Execute()176 TF_ASSIGN_OR_RETURN(std::vector<ScopedShapedBuffer> argument_buffers, in Execute()181 /*arguments=*/argument_buffers, in Execute()281 std::vector<ScopedShapedBuffer> argument_buffers; in ExecuteReplicated() local285 argument_buffers.reserve(options.num_replicas * options.arguments.size()); in ExecuteReplicated()310 argument_buffers.push_back(std::move(argument_buffer)); in ExecuteReplicated()311 argument_buffer_ptrs[index++] = &argument_buffers.back(); in ExecuteReplicated()
239 std::vector<const ShapedBuffer*> argument_buffers; in Execute() local240 argument_buffers.reserve(argument_handles.size()); in Execute()242 argument_buffers.push_back(handle->shaped_buffer()); in Execute()252 result_buffer_status = executable_->Run(argument_buffers, options); in Execute()287 std::vector<const ShapedBuffer*> argument_buffers; in ExecutePerReplica() local288 argument_buffers.reserve(argument_handles[replica].size()); in ExecutePerReplica()290 argument_buffers.push_back(handle->shaped_buffer()); in ExecutePerReplica()300 executable_->Run(argument_buffers, options); in ExecutePerReplica()