Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dreorder_ops.h27 std::vector<std::string> inputs_format; member
Dgraph_kernel_helper.cc78 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format, in BuildSelectKernelBuildInfo()
87 kernel::KernelBuildInfoPtr BuildSelectKernelBuildInfo(const std::vector<std::string> &inputs_format, in BuildSelectKernelBuildInfo()
370 auto inputs_format = build_info->GetAllInputFormats(); in CheckDefaultFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/
Dhost_kernel_metadata.cc35 std::vector<std::string> inputs_format{}; in HostMetadataInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/
Daicpu_kernel_metadata.cc62 std::vector<std::string> inputs_format{}; in AicpuMetadataInfoForSpecialNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/
Dhccl_kernel_metadata.cc95 std::vector<std::string> inputs_format{}; in HcclMetadataInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dmatmul_biasadd_fusion.cc35 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dadam_fusion.cc36 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dadam_weight_decay_fusion.cc36 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dinsert_format_transform_op.cc158 auto inputs_format = AnfAlgo::GetAllInputFormats(node); in Process() local
Dprint_reduce_fusion.cc34 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dadd_rms_norm_fusion.cc35 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dadd_cast_rms_norm_cast_fusion.cc44 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dadd_layer_norm_fusion.cc67 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dlaunch_transdata.cc54 std::vector<std::string> inputs_format = {input_format}; in SetKernelBuildInfo() local
Dkernel_select_ascend.cc78 std::vector<std::string> inputs_format(input_num, kOpFormat_DEFAULT); in GetValidFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dprint_value_type.cc31 std::vector<std::string> inputs_format; in GenerateKernelBuildInfo() local
Dinsert_format_transform_op.cc148 auto inputs_format = AnfAlgo::GetAllInputFormats(node); in InsertTransformOpForInput() local
/third_party/mindspore/mindspore-src/source/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-src/source/mindspore/ccsrc/kernel/
Dkernel_build_info.cc179 void KernelBuildInfo::SetInputsFormat(const std::vector<std::string> &inputs_format) { inputs_forma… in SetInputsFormat()
331 …nelBuildInfo::KernelBuildInfoBuilder::SetInputsFormat(const std::vector<std::string> &inputs_forma… in SetInputsFormat()
Dframework_utils.cc322 std::vector<std::string> inputs_format; in SetInputKernelBuilderInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dinsert_type_transform_op.cc218 …std::vector<std::string> inputs_format{tuple_unfold_abs->size(), builder->GetInputFormat(kIndex0)}; in CreateRealMakeTupleByTupleUnfoldInput() local
269 std::vector<std::string> inputs_format; in SetKernelInfoForNewCNode() local
322 std::vector<std::string> inputs_format = {kOpFormat_DEFAULT}; in SetKernelInfoForValueNode() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/common/
Ddevice_common_test.h182 std::vector<std::string> inputs_format; in SetKernelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dkernel_info_setter.cc371std::vector<std::string> *inputs_format, std::vector<std::string> *outputs_format, in UpdateKernelFormatInfo()
688 std::vector<std::string> inputs_format; in SetKernelInfoWithMsg() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_device_context.cc344 std::vector<std::string> inputs_format; in SetControlOpInfo() local