Searched refs:arg_buffers_ (Results 1 – 2 of 2) sorted by relevance
214 arg_buffers_.reserve(kernel->xla_input_shapes.size() + 1); in PopulateInputs()215 arg_buffers_.resize(kernel->xla_input_shapes.size()); in PopulateInputs()216 arg_ptrs_ = std::vector<ShapedBuffer*>(arg_buffers_.size()); in PopulateInputs()251 arg_buffers_[i] = absl::make_unique<ShapedBuffer>( in PopulateInputs()254 arg_buffers_[i]->set_buffer(dmem, /*index=*/{}); in PopulateInputs()255 arg_ptrs_[i] = arg_buffers_[i].get(); in PopulateInputs()
192 std::vector<std::unique_ptr<xla::ShapedBuffer>> arg_buffers_; variable