Home
last modified time | relevance | path

Searched defs:origin_data_format (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dunpack_gpu_kernel.h67 auto origin_data_format = AnfAlgo::GetOriginDataFormat(kernel_node); in Init() local
Dpack_gpu_kernel.h67 auto origin_data_format = AnfAlgo::GetOriginDataFormat(kernel_node); in Init() local
Dconcatv2_gpu_kernel.h88 auto origin_data_format = AnfAlgo::GetOriginDataFormat(kernel_node); in Init() local
Dsplit_gpu_kernel.h74 auto origin_data_format = AnfAlgo::GetOriginDataFormat(kernel_node); in Init() local
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc249 std::string *origin_data_format) { in UpdateKernelFormatInfo()
420 std::string origin_data_format = kOpFormat_DEFAULT; in SetKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dinsert_format_transform_op.cc132 auto origin_data_format = AnfAlgo::GetOriginDataFormat(node); in Process() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_build_info.cc151 …rnelBuildInfo::KernelBuildInfoBuilder::SetOriginDataFormat(const std::string &origin_data_format) { in SetOriginDataFormat()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel.h157 …int AxisTransform(const std::string &origin_data_format, const std::string &cal_format, int axis) { in AxisTransform()