Searched refs:tensor_mask (Results 1 – 3 of 3) sorted by relevance
46 const std::vector<int64_t> &tensor_mask() const { return tensors_mask_; } in tensor_mask() function
305 int64_t tensor_mask) { in ConstructRunOpParameter() argument309 if (tensor_mask == kParameterWeightTensorMask) { in ConstructRunOpParameter()
440 … std::vector<tensor::TensorPtr> *input_tensors, int64_t *const tensor_mask) { in ConvertMultiPyObjectToTensor() argument443 MS_EXCEPTION_IF_NULL(tensor_mask); in ConvertMultiPyObjectToTensor()456 *tensor_mask = kValueNodeTensorMask; in ConvertMultiPyObjectToTensor()461 … std::vector<tensor::TensorPtr> *input_tensors, int64_t *const tensor_mask) { in ConvertPyObjectToTensor() argument464 MS_EXCEPTION_IF_NULL(tensor_mask); in ConvertPyObjectToTensor()471 *tensor_mask = kValueNodeTensorMask; in ConvertPyObjectToTensor()474 *tensor_mask = kValueNodeTensorMask; in ConvertPyObjectToTensor()483 ConvertMultiPyObjectToTensor(tuple_inputs, op_prim, input_tensors, tensor_mask); in ConvertPyObjectToTensor()486 ConvertMultiPyObjectToTensor(input_object, op_prim, input_tensors, tensor_mask); in ConvertPyObjectToTensor()536 int64_t tensor_mask = op_run_info->inputs_mask[index]; in ConstructInputTensor() local[all …]