Searched refs:src_tensors_names_ (Results 1 – 5 of 5) sorted by relevance
107 src_tensors_names_(std::move(operation.src_tensors_names_)), in GPUOperation()129 src_tensors_names_ = std::move(operation.src_tensors_names_); in operator =()145 for (int i = 0; i < operation->src_tensors_names_.size(); ++i) { in AddOperation()148 src_tensors_names_.push_back(operation->src_tensors_names_[i] + in AddOperation()160 src_tensors_names_.push_back(tensor_name); in AddSrcTensor()167 src_tensors_names_.push_back(buffer_name); in AddSrcBuffer()174 src_tensors_names_.push_back(texture_name); in AddSrcTexture2D()193 src_tensors_names_.insert(src_tensors_names_.begin(), "src_tensor"); in AssembleCode()245 for (int i = 0; i < src_tensors_names_.size(); ++i) { in AddUniquePostfix()246 src_tensors_names_[i] += unique_postfix; in AddUniquePostfix()
174 std::vector<std::string> src_tensors_names_; variable
191 for (int i = 0; i < operation_->src_tensors_names_.size(); ++i) { in UpdateParams()197 RETURN_IF_ERROR(metal_args_.SetObjectRef(operation_->src_tensors_names_[i], in UpdateParams()233 metal_args_.SetObjectRef(operation_->src_tensors_names_[index], *tensor); in SetSrcTensor()
157 for (int i = 0; i < operation_->src_tensors_names_.size(); ++i) { in UpdateParams()163 RETURN_IF_ERROR(cl_args_.SetObjectRef(operation_->src_tensors_names_[i], in UpdateParams()
829 op->src_tensors_names_.push_back(std::move(name)); in Decode()864 for (auto& name : op.src_tensors_names_) { in Encode()