Home
last modified time | relevance | path

Searched defs:origin_input (Results 1 – 17 of 17) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/st/fallback/
Dtest_graph_fallback_runtime_setattr.py40 def __init__(self, origin_input): argument
66 def __init__(self, origin_input): argument
97 def __init__(self, origin_input): argument
123 def __init__(self, origin_input): argument
149 def __init__(self, origin_input): argument
178 def __init__(self, origin_input): argument
204 def __init__(self, origin_input): argument
230 def __init__(self, origin_input): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dfractional_max_pool_grad_with_fixed_ksize_impl.cu21 __global__ void Fractionalmaxpoolgradwithfixedksize(const T *origin_input, const T *out_backprop, S… in Fractionalmaxpoolgradwithfixedksize()
36 cudaError_t CalFractionalmaxpoolgradwithfixedksize(const T *origin_input, const T *out_backprop, S … in CalFractionalmaxpoolgradwithfixedksize()
Dfractionalmaxpool3dgradwithfixedksize_impl.cu30 __global__ void Fractionalmaxpool3dgradwithfixedksize(const T *origin_input, const T *out_backprop,… in Fractionalmaxpool3dgradwithfixedksize()
45 cudaError_t CalFractionalmaxpool3dgradwithfixedksize(const T *origin_input, const T *out_backprop, … in CalFractionalmaxpool3dgradwithfixedksize()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dlp_normalization_mapper.cc50 auto origin_input = cnode->inputs()[kNameLpNormInputNum]; in Mapper() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/
Dtest_fractional_max_pool_with_fixed_ksize_op.py39 def construct(self, origin_input, out_backprop, argmax): argument
Dtest_fractionalmaxpool3dwithfixedksize_op.py39 def construct(self, origin_input, out_backprop, argmax): argument
/third_party/mindspore/mindspore-src/source/tests/st/ops/
Dtest_fractionalmaxpool3dwithfixedksize.py40 def construct(self, origin_input, out_backprop, argmax): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/
Dexport_model.cc201 auto origin_input = cnode->input(i); in CloneFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dfractional_max_pool_3d_grad_with_fixed_ksize.cc39 Tensor *origin_input = ctx.Input(0); in GetInputAndCheck() local
Dfractional_max_pool_grad_with_fixed_ksize.cc57 Tensor *origin_input = ctx.Input(0); in Compute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dcore_infershape_pass.cc314 auto origin_input = return_node->inputs(); in SetSubGraphOutput() local
Dinfershape_pass.cc395 auto origin_input = return_node->inputs(); in SetSubGraphOutput() local
Dtranspose_strategy.cc106 auto origin_input = node_infer_shape->GetIntVecInput(cnode, input_index); in TransformAttrByAxes() local
Ddecrease_transpose_algo.cc681 auto origin_input = return_node->inputs(); in SetSubGraphOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dinsert_type_transform_op.cc590 bool InsertTypeTransformOp::IsInputUpdated(const AnfNodePtr &origin_input, const AnfNodePtrList &ne… in IsInputUpdated() argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
D_grad_ops.py653 def __infer__(self, origin_input, dout, mean_matrix, kernel_matrix): argument
670 def __infer__(self, origin_input, dout): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc88 static AnfNodePtr GetRealInput(const AnfNodePtr &origin_input) { in GetRealInput()