Home
last modified time | relevance | path

Searched defs:input_reduce_index (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dkernel_info_setter.cc560 std::vector<std::tuple<size_t, TypeId, TypeId>> *input_reduce_index) { in GetSelectKernelResult() argument
725 std::vector<std::tuple<size_t, TypeId, TypeId>> input_reduce_index; in SetKernelInfoWithMsg() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/
Dgpu_kernel.cc75 std::vector<ReduceDetail> input_reduce_index; in ReducePrecisionCheck() local