Searched refs:in_dtype (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/st/ops/custom_ops_tbe/ |
| D | conv_layer.py | 34 def conv_layer_cce_para_check(shape_in, shape_w, in_dtype, w_dtype, res_dtype, padh, padw, argument
43 util.check_dtype_rule(in_dtype, ['float16', 'int8', 'uint8'])
62 util.check_dtype_rule(in_dtype, ('int8',))
66 util.check_dtype_rule(in_dtype, ['float16'])
73 util.check_dtype_rule(in_dtype, ['int8', 'float16'])
76 util.check_dtype_rule(in_dtype, ['uint8', 'float16'])
86 if in_dtype == 'float16' and w_dtype == 'int8' and res_dtype == 'float16':
89 elif (in_dtype in ['int8', 'uint8'] and w_dtype == 'int8' and
93 elif (in_dtype == 'float16' and w_dtype == 'int8' and res_dtype in
97 elif (in_dtype in ['int8', 'uint8'] and w_dtype == 'int8' and res_dtype in
[all …]
|
| /third_party/mindspore/mindspore/lite/test/st/scripts/ |
| D | base_functions.sh | 7 quant_type bit_num config_file train_model in_dtype out_dtype converter_result cfg_file
50 in_dtype="DEFAULT"
65 in_dtype="FLOAT"
71 ' --inputDataType='${in_dtype}' --outputDataType='${out_dtype}' \
74 --inputDataType=${in_dtype} --outputDataType=${out_dtype} \
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_ascend.cc | 236 bool CheckHitTargetDtype(const std::map<TypeId, TypeId> &type_map, const TypeId &in_dtype, const Ty… in CheckHitTargetDtype() argument
238 auto iter = type_map.find(in_dtype); in CheckHitTargetDtype()
240 if (iter == type_map.end() && in_dtype != device_dtype) { in CheckHitTargetDtype()
245 if (iter != type_map.end() && iter->second != device_dtype && in_dtype != device_dtype) { in CheckHitTargetDtype()
248 if (in_dtype == kNumberTypeInt64 && device_dtype == kNumberTypeInt32) { in CheckHitTargetDtype()
261 auto in_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(cnode, input_index); in TagRaiseReduce() local
266 if (!CheckHitTargetDtype(type_map, in_dtype, device_dtype, &flag)) { in TagRaiseReduce()
272 auto in_dtype = AnfAlgo::GetOutputInferDataType(cnode, output_index); in TagRaiseReduce() local
278 if (!CheckHitTargetDtype(type_map, in_dtype, device_dtype, &flag)) { in TagRaiseReduce()
|
| /third_party/boost/boost/python/numpy/ |
| D | ufunc.hpp | 106 dtype in_dtype = dtype::get_builtin<TArgument>(); in call() local
108 ndarray in_array = from_object(input, in_dtype, ndarray::ALIGNED); in call()
|