Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sequence/
Dsequence_stack_gpu_kernel.cc64 auto origin_data_format = kOpFormat_DEFAULT; in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dpack_gpu_kernel.h66 auto origin_data_format = kOpFormat_DEFAULT; in Resize() local
Dconcat_gpu_kernel.cc121 std::string origin_data_format = kOpFormat_DEFAULT; in Resize() local
Dsplit_gpu_kernel.cc80 std::string origin_data_format = kOpFormat_DEFAULT; in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dinsert_format_transform_op.cc152 auto origin_data_format = AnfAlgo::GetOriginDataFormat(node); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dkernel_info_setter.cc372 std::string *origin_data_format) { in UpdateKernelFormatInfo()
709 std::string origin_data_format = kOpFormat_DEFAULT; in SetKernelInfoWithMsg() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/
Dgpu_kernel.cc147 int AxisTransform(const std::string &origin_data_format, const std::string &cal_format, int axis) { in AxisTransform()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dkernel_build_info.cc326 …rnelBuildInfo::KernelBuildInfoBuilder::SetOriginDataFormat(const std::string &origin_data_format) { in SetOriginDataFormat()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph_mgr.cc223 auto origin_data_format = kernel_info_value[kOriginDataFormat]; in LoadAnfSelectKernelBuildInfo() local