Home
last modified time | relevance | path

Searched defs:input_dtype (Results 1 – 25 of 75) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/tests/st/ops/
Dtest_ops_rms_norm.py57 def test_rms_norm_forward(mode, input_dtype): argument
97 def test_rms_norm_backward(mode, input_dtype): argument
139 def test_rms_norm_backward_cmp_with_numpy(mode, input_dtype): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dpad_ops_proto.cc39 auto input_dtype = input_desc.GetDataType(); in PadInferShapeAndType() local
116 DataType input_dtype = input_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
164 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
229 auto input_dtype = input_desc.GetDataType(); in PadV3GradInferShapeAndType() local
295 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
Dpdist_grad_proto.cc25 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
52 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in CUST_IMPLEMT_VERIFIER() local
Dmatrix_calculation_ops_proto.cc87 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
113 DataType input_dtype = op.GetInputDescByName("y_grad").GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
145 DataType input_dtype = op.GetInputDesc(0).GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
172 DataType input_dtype = op.GetInputDesc("var").GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
199 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_INFERFUNC() local
Dbroadcast_to_proto.cc42 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_INFERFUNC() local
74 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_INFERFUNC() local
Dreal_proto.cc32 DataType input_dtype = input_desc.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
Dselection_ops_proto.cc34 DataType input_dtype = x_input.GetDataType(); in InferShapeAndTypeMaskedSelect() local
62 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
134 DataType input_dtype = input_x_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
388 DataType input_dtype = op.GetInputDesc(0).GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
465 DataType input_dtype = input_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
Dbitwise_ops_proto.cc32 DataType input_dtype = op.GetInputDescByName("x").GetDataType(); in IMPLEMT_INFERFUNC() local
Dpolar_proto.cc41 DataType input_dtype = abs_desc.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
Dlstsq_proto.cc25 DataType input_dtype = op.GetInputDescByName("matrix").GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
Dconcat_proto.cc200 DataType input_dtype = input_x_desc[0].GetDataType(); in ConcatInferShapeCommon() local
210 DataType input_dtype = input_x_desc[0].GetDataType(); in ConcatInferShapeCommon() local
269 DataType input_dtype = input_x_desc[0].GetDataType(); in ConcatInferShapeCommon() local
Denviron_proto.cc54 DataType input_dtype = default_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
Dsegment_min_proto.cc69 DataType input_dtype = input_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
Dsegment_mean_proto.cc71 DataType input_dtype = input_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
Dsegment_prod_proto.cc69 DataType input_dtype = input_desc.GetDataType(); in IMPLEMT_COMMON_INFERFUNC() local
Dnn_pooling_ops_proto.cc51 DataType input_dtype = input_tensor_desc.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
69 DataType input_dtype = op.GetInputDescByName("input_grad").GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
231 DataType input_dtype = input.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
252 DataType input_dtype = input.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
272 DataType input_dtype = input.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
313 DataType input_dtype = input_desc.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
1243 DataType input_dtype = input_tensor_desc.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
1327 DataType input_dtype = input_tensor_desc.GetDataType(); in CUST_IMPLEMT_INFERFUNC() local
Dmedian_proto.cc26 DataType input_dtype = op.GetInputDescByName(input_name.c_str()).GetDataType(); in InferShapeAndTypeMedian() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_scatter_op.py29 def __init__(self, input_shape, input_dtype, use_locking): argument
40 def __init__(self, input_shape, input_dtype, use_locking): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/
Dtest_avgpool_grad_op.py57 def test_net_dtype(input_dtype): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Drandom_standard_laplace.cc113 for (const auto &input_dtype : elements_type->elements()) { in StandardLaplaceInferType() local
118 auto input_dtype = input_args[kInputIndex0]->GetType(); in StandardLaplaceInferType() local
Drandom_standard_normal.cc131 for (const auto &input_dtype : elements_type->elements()) { in RandomStandardNormalInferType() local
136 auto input_dtype = input_args[kInputIndex0]->GetType(); in RandomStandardNormalInferType() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dcast.cc51 auto input_dtype = in_tensors_.front()->data_type(); in CheckSpecs() local
Dcrop.cc40 auto input_dtype = in_tensors_.front()->data_type(); in CheckSpecsWithoutShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dstateless_dropout_genmask_replace.cc60 auto input_dtype = common::AnfAlgo::GetPrevNodeOutputInferDataType(node, 0); in Process() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_argmax_ext.cc29 TypePtr input_dtype; member

123