Home
last modified time | relevance | path

Searched defs:outputs_format (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dreorder_ops.h29 std::vector<std::string> outputs_format; member
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc48 std::vector<std::string> outputs_format; in HostMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc64 std::vector<std::string> outputs_format; in AicpuMetadataInfoForSpecialNodes() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel_metadata.cc94 std::vector<std::string> outputs_format; in HcclMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc248 … std::vector<std::string> *inputs_format, std::vector<std::string> *outputs_format, in UpdateKernelFormatInfo()
297 std::vector<std::string> outputs_format = {kOpFormat_DEFAULT}; in SetGraphKernelInfo() local
413 std::vector<std::string> outputs_format; in SetKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadd_relu_grad_v2_fusion.cc32 std::vector<std::string> outputs_format; in GenerateKernelBuildInfo() local
Dadd_relu_v2_fusion.cc32 std::vector<std::string> outputs_format; in GenerateKernelBuildInfo() local
Dmatmul_biasadd_fusion.cc32 std::vector<std::string> outputs_format; in GenerateKernelBuildInfo() local
Drelu_v2_pass.cc45 std::vector<std::string> outputs_format; in GenerateKernelBuildInfo() local
Dinsert_format_transform_op.cc150 auto outputs_format = AnfAlgo::GetAllOutputFormats(node); in Process() local
Dadam_fusion.cc32 std::vector<std::string> outputs_format; in GenerateKernelBuildInfo() local
Dadam_weight_decay_fusion.cc32 std::vector<std::string> outputs_format; in GenerateKernelBuildInfo() local
Dprint_reduce_fusion.cc31 std::vector<std::string> outputs_format; in GenerateKernelBuildInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_transdata.cc109 std::vector<std::string> outputs_format = {dst_format_}; in ConstructKernelGraphAndSetAttr() local
Dkernel_select_graph_kernel.cc297 std::vector<std::string> outputs_format = {(*graph_input_format)[i]}; in UpdateInputsKernelInfo() local
417 std::vector<std::string> outputs_format = {graph_input_format[i]}; in UpdateFormatsAndDtypes() local
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc43 const std::vector<std::string> &outputs_format, in CreateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/
Dinsert_format_transform_op.cc137 auto outputs_format = AnfAlgo::GetAllOutputFormats(node); in InsertTransformOpForOutput() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.cc161 …lBuildInfo::KernelBuildInfoBuilder::SetOutputsFormat(const std::vector<std::string> &outputs_forma… in SetOutputsFormat()
Dcommon_utils.cc365 std::vector<std::string> outputs_format; in SetOutputKernelBuilderInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc154 std::vector<std::string> outputs_format; in SetControlOpInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc122 std::vector<std::string> outputs_format; in GetCommonPatternKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc121 std::vector<std::string> outputs_format; in CreateFusionOpKernelInfo() local