Searched defs:in_dtype (Results 1 – 3 of 3) 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
150 def conv_layer_cce(shape_in, shape_w, in_dtype, w_dtype, res_dtype, padh, padw, strideh, stridew, argument
|
| /third_party/boost/boost/python/numpy/ |
| D | ufunc.hpp | 106 dtype in_dtype = dtype::get_builtin<TArgument>(); in call() local
|
| /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()
261 auto in_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(cnode, input_index); in TagRaiseReduce() local
272 auto in_dtype = AnfAlgo::GetOutputInferDataType(cnode, output_index); in TagRaiseReduce() local
|