Home
last modified time | relevance | path

Searched defs:outputs_type (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dreorder_ops.h31 std::vector<TypeId> outputs_type; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc49 std::vector<TypeId> outputs_type; in HostMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dreplace_addn_fusion.cc43 std::vector<TypeId> outputs_type; in Process() local
Dreplace_momentum_cast_fusion.cc47 std::vector<TypeId> outputs_type; in Process() local
Dbatch_norm_relu_fusion.cc80 std::vector<TypeId> outputs_type; in Process() local
Dbatch_norm_add_relu_fusion.cc85 std::vector<TypeId> outputs_type; in Process() local
Dpost_batch_norm_add_relu_fusion.cc85 std::vector<TypeId> outputs_type; in Process() local
Dbatch_norm_relu_grad_fusion.cc97 std::vector<TypeId> outputs_type; in Process() local
Dadd_relu_grad_v2_fusion.cc34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
Dadd_relu_v2_fusion.cc34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
Dmatmul_biasadd_fusion.cc34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
Drelu_v2_pass.cc47 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
Dbatch_norm_add_relu_grad_fusion.cc70 std::vector<TypeId> outputs_type; in SetShapeAndType() local
Dadam_fusion.cc34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
Dprint_reduce_fusion.cc33 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
Dadam_weight_decay_fusion.cc34 std::vector<TypeId> outputs_type; in GenerateKernelBuildInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc65 std::vector<TypeId> outputs_type; in AicpuMetadataInfoForSpecialNodes() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel_metadata.cc95 std::vector<TypeId> outputs_type; in HcclMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc345 const std::vector<TypeId> &outputs_type) { in PrintUnsupportedTypeException()
414 std::vector<TypeId> outputs_type; in SetKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc155 std::vector<TypeId> outputs_type; in SetControlOpInfo() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc201 std::vector<TypePtr> outputs_type; in ExtractOutputTypeByNode() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Doperator_info.h83 const std::vector<TypePtr> &outputs_type() const { return outputs_type_; } in outputs_type() function
Doperator_info.cc1683 Status OperatorInfo::set_outputs_type(const std::vector<TypePtr> &outputs_type) { in set_outputs_type()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_auto_parallel.cc311 auto outputs_type = ExtractOutputTypeByNode(cnode); in CreateTheOperatorInfo() local