Home
last modified time | relevance | path

Searched defs:tensor_types (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dfractional_max_pool_grad.cc75 std::map<std::string, TypePtr> tensor_types; in FractionalMaxPoolGradInferType() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dsparse_apply_momentum.cc70 std::map<std::string, Tensor *> tensor_types; in ValidParam() local
Dsparse_apply_proximal_gradient_descent.cc70 std::map<std::string, Tensor *> tensor_types; in ValidParam() local
Dsparse_apply_adagrad_da.cc107 std::map<std::string, Tensor *> tensor_types; in ValidParam() local
Dsparse_apply_centered_rms_prop.cc82 std::map<std::string, Tensor *> tensor_types; in ValidParam() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/op_declare/
Dop_proto.cc82 …ctor<enum ge::DataType> ParseGeTypes(const std::string &op_name, const std::string &tensor_types) { in ParseGeTypes()