Searched refs:metal_args_ (Results 1 – 2 of 2) sorted by relevance
83 RETURN_IF_ERROR(metal_args_.Init(linkables, device, &operation_->args_, in Compile()197 RETURN_IF_ERROR(metal_args_.SetObjectRef(operation_->src_tensors_names_[i], in UpdateParams()206 RETURN_IF_ERROR(metal_args_.SetObjectRef(operation_->dst_tensors_names_[i], in UpdateParams()209 RETURN_IF_ERROR(operation_->BindArguments(&metal_args_)); in UpdateParams()219 metal_args_.Encode(encoder, 0); in Encode()233 metal_args_.SetObjectRef(operation_->src_tensors_names_[index], *tensor); in SetSrcTensor()239 metal_args_.SetObjectRef(operation_->dst_tensors_names_[index], *tensor); in SetDstTensor()
77 MetalArguments metal_args_; variable