Searched refs:kPrimTensorSummary (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | tensor_summary.h | 35 TensorSummary() : PrimitiveC(prim::kPrimTensorSummary->name()) {} in TensorSummary()
|
| D | tensor_summary.cc | 53 REGISTER_PRIMITIVE_EVAL_IMPL(TensorSummary, prim::kPrimTensorSummary, TensorSummaryInfer, nullptr, …
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | anf_extends.cc | 45 …if (IsApply(prim::kPrimScalarSummary) || IsApply(prim::kPrimTensorSummary) || IsApply(prim::kPrimI… in fullname_with_scope()
|
| D | anf.cc | 397 …IsPrimitiveCNode(node, prim::kPrimTensorSummary) || IsPrimitiveCNode(node, prim::kPrimHistogramSum… in GetVirtualNodeTargetFromInputs()
530 …IsPrimitiveCNode(node, prim::kPrimTensorSummary) || IsPrimitiveCNode(node, prim::kPrimHistogramSum… in GetOriginNodeTarget()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | array_ops_declare.cc | 38 REG_ADPT_DESC(TensorSummary, prim::kPrimTensorSummary->name(), ADPT_DESC(Summary))
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | branch_culling.cc | 74 … {prim::kPrimTensorSummary, {1}}, in InConvertWhiteList()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | graph_partition.cc | 465 …IsPrimitiveCNode(node, prim::kPrimTensorSummary) || IsPrimitiveCNode(node, prim::kPrimHistogramSum… in RemoveUselessDependency()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 542 auto prim = prim::kPrimTensorSummary; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/common/ |
| D | gllo_utils.cc | 163 IsPrimitive(input, prim::kPrimTensorSummary) || in IsRealKernel()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 563 inline const PrimitivePtr kPrimTensorSummary = std::make_shared<Primitive>("TensorSummary"); variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 367 … (IsPrimitiveCNode(n, prim::kPrimScalarSummary) || IsPrimitiveCNode(n, prim::kPrimTensorSummary) || in ExistSummaryNode()
1826 … (IsPrimitiveCNode(n, prim::kPrimScalarSummary) || IsPrimitiveCNode(n, prim::kPrimTensorSummary) || in SetSummaryNodes()
|
| D | anf_runtime_algorithm.cc | 63 …prim::kPrimImageSummary, prim::kPrimScalarSummary, prim::kPrimTensorSummary, prim::kPrimHistogramS… in IsRealKernelCNode()
|