Searched refs:arg_ptrs_ (Results 1 – 2 of 2) sorted by relevance
185 const std::vector<xla::ShapedBuffer*>& arguments() const { return arg_ptrs_; } in arguments()193 std::vector<xla::ShapedBuffer*> arg_ptrs_; variable
216 arg_ptrs_ = std::vector<ShapedBuffer*>(arg_buffers_.size()); in PopulateInputs()243 arg_ptrs_[i] = const_cast<ShapedBuffer*>(&xla_tensor->shaped_buffer()); in PopulateInputs()255 arg_ptrs_[i] = arg_buffers_[i].get(); in PopulateInputs()