Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dreorder_ops.h28 std::vector<std::string> inputs_format; member
Dgraph_kernel_helper.cc287 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format, in BuildSelectKernelBuildInfo()
303 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format, in BuildSelectKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/
Dhost_kernel_metadata.cc41 std::vector<std::string> inputs_format{}; in HostMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_metadata.cc54 std::vector<std::string> inputs_format{}; in AicpuMetadataInfoForSpecialNodes() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/
Dhccl_kernel_metadata.cc87 std::vector<std::string> inputs_format{}; in HcclMetadataInfo() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_launch_transdata.cc108 std::vector<std::string> inputs_format = {src_format_}; in ConstructKernelGraphAndSetAttr() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dadd_relu_grad_v2_fusion.cc31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dadd_relu_v2_fusion.cc31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dmatmul_biasadd_fusion.cc31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Drelu_v2_pass.cc44 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dinsert_format_transform_op.cc138 auto inputs_format = AnfAlgo::GetAllInputFormats(node); in Process() local
Dadam_fusion.cc31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dprint_reduce_fusion.cc30 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dadam_weight_decay_fusion.cc31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dcommon_subexpression_elimination_test.cc42 kernel::KernelBuildInfoPtr CreateKernelBuildInfo(const std::vector<std::string> &inputs_format, in CreateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc248std::vector<std::string> *inputs_format, std::vector<std::string> *outputs_format, in UpdateKernelFormatInfo()
406 std::vector<std::string> inputs_format; in SetKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/
Dinsert_format_transform_op.cc113 auto inputs_format = AnfAlgo::GetAllInputFormats(node); in InsertTransformOpForInput() local
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/cpu/
Dcpu_device_context.cc147 std::vector<std::string> inputs_format; in SetControlOpInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.cc156 …nelBuildInfo::KernelBuildInfoBuilder::SetInputsFormat(const std::vector<std::string> &inputs_forma… in SetInputsFormat()
Dcommon_utils.cc306 std::vector<std::string> inputs_format; in SetInputKernelBuilderInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/
Dtbe_kernel_select.cc108 std::vector<std::string> inputs_format; in GetCommonPatternKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/
Dub_pattern_fusion.cc113 std::vector<std::string> inputs_format; in CreateFusionOpKernelInfo() local