!45334 fix some issues master
Merge pull request !45334 from 李林杰/1109_fix_some_issues_master
This commit is contained in:
commit
2d0280c0eb
|
@ -472,7 +472,7 @@ dnnl::memory::data_type MKLCpuKernelMod::GetDnnlDataType(TypeId ms_type_id) cons
|
|||
{kNumberTypeUInt8, dnnl::memory::data_type::u8}};
|
||||
auto iter = dnnl_data_type_map.find(ms_type_id);
|
||||
if (iter == dnnl_data_type_map.end()) {
|
||||
MS_LOG(ERROR) << "Dnnl do not support data type:" << TypeIdToString(ms_type_id);
|
||||
MS_LOG(WARNING) << "Dnnl do not support data type:" << TypeIdToString(ms_type_id);
|
||||
return dnnl::memory::data_type::undef;
|
||||
}
|
||||
return iter->second;
|
||||
|
|
|
@ -24,6 +24,7 @@ strided_slice_grad_d_op_info = TBERegOp("StridedSliceGrad") \
|
|||
.kernel_name("strided_slice_grad") \
|
||||
.partial_flag(True) \
|
||||
.dynamic_compile_static(True) \
|
||||
.need_check_supported(True) \
|
||||
.attr("begin_mask", "optional", "int", "all", "0") \
|
||||
.attr("end_mask", "optional", "int", "all", "0") \
|
||||
.attr("ellipsis_mask", "optional", "int", "all", "0") \
|
||||
|
|
Loading…
Reference in New Issue