Searched refs:num_segments_tensor (Results 1 – 1 of 1) sorted by relevance
324 auto num_segments_tensor = num_segments_value_ptr->cast<tensor::TensorPtr>(); in GetUnsortedSegmentOpScalarArg() local325 MS_EXCEPTION_IF_NULL(num_segments_tensor); in GetUnsortedSegmentOpScalarArg()327 num_segments_value = *static_cast<int64_t *>(num_segments_tensor->data_c()); in GetUnsortedSegmentOpScalarArg()329 num_segments_value = *static_cast<int32_t *>(num_segments_tensor->data_c()); in GetUnsortedSegmentOpScalarArg()