Searched refs:dst_tensors_names_ (Results 1 – 5 of 5) sorted by relevance
108 dst_tensors_names_(std::move(operation.dst_tensors_names_)), in GPUOperation()130 dst_tensors_names_ = std::move(operation.dst_tensors_names_); in operator =()151 for (int i = 0; i < operation->dst_tensors_names_.size(); ++i) { in AddOperation()152 dst_tensors_names_.push_back(operation->dst_tensors_names_[i] + in AddOperation()181 dst_tensors_names_.push_back(tensor_name); in AddDstTensor()201 dst_tensors_names_.insert(dst_tensors_names_.begin(), "dst_tensor"); in AssembleCode()248 for (int i = 0; i < dst_tensors_names_.size(); ++i) { in AddUniquePostfix()249 dst_tensors_names_[i] += unique_postfix; in AddUniquePostfix()
175 std::vector<std::string> dst_tensors_names_; variable
82 {operation_->dst_tensors_names_[0], operation_->elementwise_code_}}; in Compile()200 for (int i = 0; i < operation_->dst_tensors_names_.size(); ++i) { in UpdateParams()206 RETURN_IF_ERROR(metal_args_.SetObjectRef(operation_->dst_tensors_names_[i], in UpdateParams()239 metal_args_.SetObjectRef(operation_->dst_tensors_names_[index], *tensor); in SetDstTensor()
166 for (int i = 0; i < operation_->dst_tensors_names_.size(); ++i) { in UpdateParams()172 RETURN_IF_ERROR(cl_args_.SetObjectRef(operation_->dst_tensors_names_[i], in UpdateParams()190 {{operation_->dst_tensors_names_[0], operation_->elementwise_code_}}, in Compile()
833 op->dst_tensors_names_.push_back(std::move(name)); in Decode()870 for (auto& name : op.dst_tensors_names_) { in Encode()