Home
last modified time | relevance | path

Searched refs:CheckInteger (Results 1 – 25 of 150) sorted by relevance

123456

/third_party/mindspore/mindspore/core/ops/
Dconv2d_transpose.cc47 … MakeValue(CheckAndConvertUtils::CheckInteger(kInChannel, in_channel, kGreaterThan, 0, name()))); in set_in_channel()
52 … MakeValue(CheckAndConvertUtils::CheckInteger(kOutChannel, out_channel, kGreaterThan, 0, name()))); in set_out_channel()
57 …(void)CheckAndConvertUtils::CheckInteger(kKernelSize, SizeToLong(kernel_size.size()), kEqual, kern… in set_kernel_size()
59 (void)CheckAndConvertUtils::CheckInteger(kKernelSize, item, kGreaterEqual, 1, name()); in set_kernel_size()
66 …(void)CheckAndConvertUtils::CheckInteger(kStride, SizeToLong(stride.size()), kEqual, stride_size, … in set_stride()
68 (void)CheckAndConvertUtils::CheckInteger(kStride, item, kGreaterEqual, 1, name()); in set_stride()
75 …(void)CheckAndConvertUtils::CheckInteger(kDilation, SizeToLong(dilation.size()), kGreaterEqual, di… in set_dilation()
95 …(void)CheckAndConvertUtils::CheckInteger("pad_size", SizeToLong(pad.size()), kEqual, pad_size, nam… in set_pad()
100 …(void)AddAttr(kMode, MakeValue(CheckAndConvertUtils::CheckInteger(kMode, mode, kEqual, 1, name()))… in set_mode()
104 …(void)AddAttr(kGroup, MakeValue(CheckAndConvertUtils::CheckInteger(kGroup, group, kGreaterThan, 0,… in set_group()
[all …]
Dbatch_to_space.cc51 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, 1,… in BatchToSpaceInfer()
64 …(void)CheckAndConvertUtils::CheckInteger("x rank", SizeToLong(x_shape.size()), kEqual, x_rank, pri… in BatchToSpaceInfer()
65 …(void)CheckAndConvertUtils::CheckInteger("block_size size", SizeToLong(block_size.size()), kEqual,… in BatchToSpaceInfer()
67 …(void)CheckAndConvertUtils::CheckInteger("crops size", SizeToLong(crops.size()), kEqual, attr_size… in BatchToSpaceInfer()
68 …(void)CheckAndConvertUtils::CheckInteger("crops[0] size", SizeToLong(crops[0].size()), kEqual, att… in BatchToSpaceInfer()
69 …(void)CheckAndConvertUtils::CheckInteger("crops[1] size", SizeToLong(crops[1].size()), kEqual, att… in BatchToSpaceInfer()
76 (void)CheckAndConvertUtils::CheckInteger("x_shape[0] % (block_size[0]*block_size[1])", in BatchToSpaceInfer()
Dreverse_sequence.cc43 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, i… in ReverseSequenceInfer()
53 …(void)CheckAndConvertUtils::CheckInteger("seq_dim", seq_dim, kLessEqual, SizeToLong(input_shape.si… in ReverseSequenceInfer()
54 …(void)CheckAndConvertUtils::CheckInteger("batch_dim", batch_dim, kLessEqual, SizeToLong(input_shap… in ReverseSequenceInfer()
56 (void)CheckAndConvertUtils::CheckInteger("batch_dim", batch_dim, kNotEqual, seq_dim, prim_name); in ReverseSequenceInfer()
57 …(void)CheckAndConvertUtils::CheckInteger("seq_lengths rank", SizeToLong(seq_lengths.size()), kEqua… in ReverseSequenceInfer()
58 (void)CheckAndConvertUtils::CheckInteger("seq_lengths vector size", seq_lengths[0], kEqual, in ReverseSequenceInfer()
Dlstm.cc39 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in LstmInfer()
46 …(void)CheckAndConvertUtils::CheckInteger("x_shape.size()", SizeToLong(x_input_shape.size()), kEqua… in LstmInfer()
48 …(void)CheckAndConvertUtils::CheckInteger("x_shape[2]", x_input_shape[2], kEqual, input_x_size, pri… in LstmInfer()
50 …(void)CheckAndConvertUtils::CheckInteger("h_shape.size()", SizeToLong(h_input_shape.size()), kEqua… in LstmInfer()
58 …(void)CheckAndConvertUtils::CheckInteger("h_shape[0]", h_input_shape[0], kEqual, num_layers * num_… in LstmInfer()
60 …(void)CheckAndConvertUtils::CheckInteger("h_shape[1]", h_input_shape[1], kEqual, x_input_shape[1],… in LstmInfer()
61 …(void)CheckAndConvertUtils::CheckInteger("h_shape[2]", h_input_shape[2], kEqual, hidden_size, prim… in LstmInfer()
107 (void)CheckAndConvertUtils::CheckInteger(kInput_size, input_size, kGreaterThan, 0, this->name()); in set_input_size()
112 …(void)CheckAndConvertUtils::CheckInteger(kHidden_size, hidden_size, kGreaterThan, 0, this->name()); in set_hidden_size()
117 (void)CheckAndConvertUtils::CheckInteger(kNumLayers, num_layers, kGreaterThan, 0, this->name()); in set_num_layers()
Dlsh_projection.cc41 …(void)CheckAndConvertUtils::CheckInteger("input0 rank", SizeToLong(input0.size()), kEqual, input0_… in LshProjectionInfer()
42 …(void)CheckAndConvertUtils::CheckInteger("input0_shape_dimen_1", input0[1], kLessEqual, input0_las… in LshProjectionInfer()
43 …(void)CheckAndConvertUtils::CheckInteger("input1 rank", SizeToLong(input1.size()), kGreaterEqual, … in LshProjectionInfer()
47 …(void)CheckAndConvertUtils::CheckInteger("input2 rank", SizeToLong(input2.size()), kEqual, 1, op_n… in LshProjectionInfer()
48 …(void)CheckAndConvertUtils::CheckInteger("input2_shape_dimen_0", input2[0], kEqual, input1[0], op_… in LshProjectionInfer()
Dctcloss.cc33 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kGreaterE… in CheckCTCLossInputs()
49 …(void)CheckAndConvertUtils::CheckInteger("inputs rank", SizeToLong(inputs_shape.size()), kEqual, i… in CheckCTCLossInputs()
50 …(void)CheckAndConvertUtils::CheckInteger("label_indices rank", SizeToLong(labels_indices_shape.siz… in CheckCTCLossInputs()
52 …(void)CheckAndConvertUtils::CheckInteger("label_indices second dim", labels_indices_shape[1], kEqu… in CheckCTCLossInputs()
54 …(void)CheckAndConvertUtils::CheckInteger("label_values rank", int64_t(labels_values_shape.size()),… in CheckCTCLossInputs()
56 …(void)CheckAndConvertUtils::CheckInteger("sequence_length rank", int64_t(sequence_length_shape.siz… in CheckCTCLossInputs()
Dapply_adagrad_d_a.cc33 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferShape()
47 …(void)CheckAndConvertUtils::CheckInteger("lr_shape size", SizeToInt(lr_shape.size()), kEqual, inpu… in InferShape()
49 …(void)CheckAndConvertUtils::CheckInteger("l1_shape size", SizeToInt(l1_shape.size()), kEqual, inpu… in InferShape()
51 …(void)CheckAndConvertUtils::CheckInteger("l2_shape size", SizeToInt(l2_shape.size()), kEqual, inpu… in InferShape()
53 …(void)CheckAndConvertUtils::CheckInteger("global_step_shape size", SizeToInt(global_step_shape.siz… in InferShape()
63 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in InferType()
Dsparse_apply_r_m_s_prop.cc33 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, 6,… in InferShape()
48 …(void)CheckAndConvertUtils::CheckInteger("size of lr_shape", SizeToLong(lr_shape.size()), kEqual, … in InferShape()
60 …(void)CheckAndConvertUtils::CheckInteger("indices dim", SizeToLong(indices_shape.size()), kEqual, … in InferShape()
63 …(void)CheckAndConvertUtils::CheckInteger("dimension of var", SizeToLong(var_shape.size()), kGreate… in InferShape()
75 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, 6,… in InferType()
Dspace_to_batch_nd.cc34 …(void)CheckAndConvertUtils::CheckInteger("input_x rank", SizeToLong(x_shape.size()), kEqual, shape… in InferShape()
65 …(void)CheckAndConvertUtils::CheckInteger(kPaddings, SizeToLong(paddings.size()), kEqual, pad_size,… in set_paddings()
72 …(void)CheckAndConvertUtils::CheckInteger(kPaddings, paddings[i][j], kGreaterEqual, 0LL, this->name… in set_paddings()
84 …(void)CheckAndConvertUtils::CheckInteger(kBlockShape, SizeToLong(block_shape.size()), kEqual, bloc… in set_block_shape()
87 …(void)CheckAndConvertUtils::CheckInteger(kBlockShape, block_shape[i], kGreaterEqual, 1LL, this->na… in set_block_shape()
Dbatch_to_space_nd.cc34 …(void)CheckAndConvertUtils::CheckInteger("input_x rank", SizeToLong(x_shape.size()), kEqual, x_ran… in InferShape()
67 …(void)CheckAndConvertUtils::CheckInteger(kCrops, SizeToLong(crops.size()), kEqual, crop_size, this… in set_crops()
74 (void)CheckAndConvertUtils::CheckInteger(kCrops, crops[i][j], kGreaterEqual, 0, this->name()); in set_crops()
86 …(void)CheckAndConvertUtils::CheckInteger(kBlockShape, SizeToLong(block_shape.size()), kEqual, bloc… in set_block_shape()
89 …(void)CheckAndConvertUtils::CheckInteger(kBlockShape, block_shape[i], kGreaterEqual, 1, this->name… in set_block_shape()
Dbias_add.cc39 …(void)CheckAndConvertUtils::CheckInteger("arg size", SizeToLong(input_args.size()), kEqual, input_… in InferShape()
49 …(void)CheckAndConvertUtils::CheckInteger("bias rank", SizeToLong(bias_shape.size()), kEqual, 1, pr… in InferShape()
51 …(void)CheckAndConvertUtils::CheckInteger("x rank", SizeToLong(input_shape.size()), kGreaterEqual, … in InferShape()
80 …(void)CheckAndConvertUtils::CheckInteger("biasadd_infer", SizeToLong(input_args.size()), kEqual, i… in InferType()
Ddepth_to_space.cc50 …(void)CheckAndConvertUtils::CheckInteger("input number", int64_t(input_args.size()), kEqual, 1, pr… in DepthToSpaceInfer()
63 …(void)CheckAndConvertUtils::CheckInteger("x rank", SizeToLong(x_shape.size()), kEqual, x_rank, pri… in DepthToSpaceInfer()
65 (void)CheckAndConvertUtils::CheckInteger("x_shape[1] % (block_size*block_size)", in DepthToSpaceInfer()
Dfake_quant_with_min_max_vars_per_channel.cc27 (void)CheckAndConvertUtils::CheckInteger(kNumBits, num_bits, kGreaterThan, 0, this->name()); in set_num_bits()
50 …(void)CheckAndConvertUtils::CheckInteger("x rank", (int64_t)x_shape.size(), kGreaterThan, 1, op_na… in FakeQuantWithMinMaxVarsPerChannelInfer()
52 …(void)CheckAndConvertUtils::CheckInteger("min shape", (int64_t)min_shape.size(), kEqual, 1, op_nam… in FakeQuantWithMinMaxVarsPerChannelInfer()
Dstrided_slice.cc71 (void)CheckAndConvertUtils::CheckInteger("infer", SizeToLong(new_axis_pos.size()), kGreaterEqual, in EllipsisInferShape()
148 …(void)CheckAndConvertUtils::CheckInteger("end_v size", SizeToLong(end_v.size()), kGreaterEqual, Si… in ComputeInferShape()
150 …(void)CheckAndConvertUtils::CheckInteger("strides_v size", SizeToLong(strides_v.size()), kGreaterE… in ComputeInferShape()
232 (void)CheckAndConvertUtils::CheckInteger(kBeginMask, begin_mask, kGreaterEqual, 0, this->name()); in set_begin_mask()
240 (void)CheckAndConvertUtils::CheckInteger(kEndMask, end_mask, kGreaterEqual, 0, this->name()); in set_end_mask()
248 …(void)CheckAndConvertUtils::CheckInteger(kEllipsisMask, ellipsis_mask, kGreaterEqual, 0, this->nam… in set_ellipsis_mask()
262 …(void)CheckAndConvertUtils::CheckInteger(kNewAxisMask, new_axis_mask, kGreaterEqual, 0, this->name… in set_new_axis_mask()
270 …(void)CheckAndConvertUtils::CheckInteger(kShrinkAxisMask, shrink_axis_mask, kGreaterEqual, 0, this… in set_shrink_axis_mask()
Dconcat.cc36 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, 1,… in ConcatInfer()
44 …(void)CheckAndConvertUtils::CheckInteger("concat element num", SizeToLong(elements.size()), kGreat… in ConcatInfer()
62 …(void)CheckAndConvertUtils::CheckInteger(elementi + " shape rank", SizeToLong(elementi_shape.size(… in ConcatInfer()
Dindex_add.cc29 …(void)CheckAndConvertUtils::CheckInteger("input numbers", SizeToLong(input_args.size()), kEqual, i… in IndexAddInferShape()
43 (void)CheckAndConvertUtils::CheckInteger("idx size", idx_rank, kEqual, 1, prim_name); in IndexAddInferShape()
62 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kEqual, in… in IndexAddInferType()
Dtranspose.cc56 (void)CheckAndConvertUtils::CheckInteger("perm element", i, kGreaterEqual, 0, op_name); in InferShape()
57 …(void)CheckAndConvertUtils::CheckInteger("perm element", i, kLessThan, SizeToLong(p_value.size()),… in InferShape()
93 …(void)CheckAndConvertUtils::CheckInteger("Transpose infer", SizeToLong(input_args.size()), kGreate… in TransposeInfer()
Dspace_to_batch.cc34 …(void)CheckAndConvertUtils::CheckInteger("x rank", SizeToLong(input_shape.size()), kEqual, x_rank,… in InferShape()
42 …(void)CheckAndConvertUtils::CheckInteger("padded shape", padded % SizeToLong(block_shape_vector.si… in InferShape()
67 …(void)CheckAndConvertUtils::CheckInteger(kPadding, paddings[i][j], kGreaterEqual, 0, this->name()); in set_paddings()
Daddn.cc36 …(void)CheckAndConvertUtils::CheckInteger("concat element num", SizeToLong(elements.size()), kGreat… in AddNInferShape()
74 …(void)CheckAndConvertUtils::CheckInteger("concat element num", SizeToLong(elements.size()), kGreat… in AddNInferType()
95 …(void)CheckAndConvertUtils::CheckInteger("input number", SizeToLong(input_args.size()), kGreaterEq… in AddNInfer()
/third_party/mindspore/mindspore/core/ops/fusion/
Dconv2d_transpose_fusion.cc44 …(void)CheckAndConvertUtils::CheckInteger(kKernelSize, SizeToLong(kernel_size.size()), kEqual, kern… in set_kernel_size()
46 (void)CheckAndConvertUtils::CheckInteger(kKernelSize, item, kGreaterEqual, 1, name()); in set_kernel_size()
53 …(void)CheckAndConvertUtils::CheckInteger(kDilation, SizeToLong(dilation.size()), kEqual, dilation_… in set_dilation()
55 (void)CheckAndConvertUtils::CheckInteger(kDilation, item, kGreaterEqual, 1, name()); in set_dilation()
61 …(void)CheckAndConvertUtils::CheckInteger(kOutputPaddings, SizeToLong(output_paddings.size()), kGre… in set_output_paddings()
64 (void)CheckAndConvertUtils::CheckInteger(kOutputPaddings, item, kGreaterEqual, 0, name()); in set_output_paddings()
/third_party/mindspore/mindspore/core/ops/grad/
Dstrided_slice_grad.cc36 (void)CheckAndConvertUtils::CheckInteger(kBeginMask, begin_mask, kGreaterEqual, 0, this->name()); in set_begin_mask()
45 (void)CheckAndConvertUtils::CheckInteger(kEndMask, end_mask, kGreaterEqual, 0, this->name()); in set_end_mask()
54 …(void)CheckAndConvertUtils::CheckInteger(kEllipsisMask, ellipsis_mask, kGreaterEqual, 0, this->nam… in set_ellipsis_mask()
69 …(void)CheckAndConvertUtils::CheckInteger(kNewAxisMask, new_axis_mask, kGreaterEqual, 0, this->name… in set_new_axis_mask()
78 …(void)CheckAndConvertUtils::CheckInteger(kShrinkAxisMask, shrink_axis_mask, kGreaterEqual, 0, this… in set_shrink_axis_mask()
Davg_pool_3d_grad.cc34 …(void)CheckAndConvertUtils::CheckInteger("input size", SizeToLong(input_args.size()), kEqual, inpu… in InferShape()
39 …(void)CheckAndConvertUtils::CheckInteger("grad_rank", SizeToLong(grad_shape.size()), kEqual, k5DIn… in InferShape()
53 …(void)CheckAndConvertUtils::CheckInteger("input size", SizeToLong(input_args.size()), kEqual, inpu… in InferType()
Dlstm_grad.cc35 (void)CheckAndConvertUtils::CheckInteger(kInput_size, input_size, kGreaterThan, 0, this->name()); in set_input_size()
40 …(void)CheckAndConvertUtils::CheckInteger(kHidden_size, hidden_size, kGreaterThan, 0, this->name()); in set_hidden_size()
45 (void)CheckAndConvertUtils::CheckInteger(kNumLayers, num_layers, kGreaterThan, 0, this->name()); in set_num_layers()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dpower_fp16.c60 if (CheckInteger(*exponent)) { in PowerBroadCastFp16()
97 …PowerScalarFunFp16_ = CheckInteger(exponent[i + j]) ? OptimizedPowerScalarFp16 : StdPowerScalarFp1… in PowerSingleFp16()
103 PowerScalarFunFp16_ = CheckInteger(exponent[i]) ? OptimizedPowerScalarFp16 : StdPowerScalarFp16; in PowerSingleFp16()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dpower_fp32.c57 if (CheckInteger(*exponent)) { in PowerBroadCast()
93 PowerScalarFun_ = CheckInteger(exponent[i + j]) ? OptimizedPowerScalar : StdPowerScalar; in PowerSingle()
99 PowerScalarFun_ = CheckInteger(exponent[i]) ? OptimizedPowerScalar : StdPowerScalar; in PowerSingle()

123456