Home
last modified time | relevance | path

Searched refs:CheckTensorDType (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc238 …(void)CheckTensorDType(x, {kFloat16, kFloat32, kFloat64, kInt32}, "Input 0 (x) for UnsortedSegment… in InferImplUnsortedSegmentSum()
239 …(void)CheckTensorDType(segment_ids, {kInt32, kInt64}, "Input 1 (segment_ids) for UnsortedSegmentSu… in InferImplUnsortedSegmentSum()
293 …(void)CheckTensorDType(x, {kFloat16, kFloat32, kInt32}, "Input 0 (x) for UnsortedSegmentMax should… in InferImplUnsortedSegmentMax()
294 …(void)CheckTensorDType(segment_ids, {kInt32, kInt64}, "Input 1 (segment_ids) for UnsortedSegmentMa… in InferImplUnsortedSegmentMax()
345 …(void)CheckTensorDType(x, {kFloat16, kFloat32, kInt32}, "Input 0 (x) for UnsortedSegmentMin should… in InferImplUnsortedSegmentMin()
346 …(void)CheckTensorDType(segment_ids, {kInt32}, "Input 1 (segment_ids) for UnsortedSegmentMin should… in InferImplUnsortedSegmentMin()
877 …(void)CheckTensorDType(lengths, {kInt32, kInt64}, "Input 1 (lengths) for SequenceMask should be on… in InferImplSequenceMask()
1026 …TypePtr range_start_type = CheckTensorDType(range_start, supported_types, "range_start input of Ra… in InferImplRange()
1027 …TypePtr range_end_type = CheckTensorDType(range_end, supported_types, "range_start input of Range … in InferImplRange()
1028 …TypePtr range_delta_type = CheckTensorDType(range_delta, supported_types, "range_start input of Ra… in InferImplRange()
[all …]
Dparam_validator.cc52 TypePtr CheckTensorDType(const AbstractTensorPtr &tensor, const TypePtrList &accepts, in CheckTensorDType() function
97 return CheckTensorDType(sample_tensor, accepts, error_message_prefix); in CheckTensorsDTypeSame()
Dprim_nn.cc47 (void)CheckTensorDType(input_tensor, {kFloat16, kFloat32}, "Input 0 of Pooling should be %s"); in InferImplPooling()
142 (void)CheckTensorDType(input_tensor, {kFloat16, kFloat32}, "param x of BatchNorm should be"); in InferImplBatchNorm()
405 …(void)CheckTensorDType(tensor, {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}, "Input of HSig… in InferImplHSigmoid()
416 (void)CheckTensorDType(dout, {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}, in InferImplHSigmoidGrad()
418 …(void)CheckTensorDType(x, {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}, "X of HSigmoidGrad … in InferImplHSigmoidGrad()
Dparam_validator.h35 TypePtr CheckTensorDType(const AbstractTensorPtr &tensor, const TypePtrList &accepts,
Dprim_maths.cc231 (void)CheckTensorDType(start, {kFloat32}, "Input 0 (start) for LinSpace should be %s"); in InferImplLinSpace()
232 (void)CheckTensorDType(stop, {kFloat32}, "Input 1 (stop) for LinSpace should be %s"); in InferImplLinSpace()