Searched defs:src_dtype (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/ |
| D | _basic.py | 85 def _shape_check(shape_a, shape_b, shape_bias, src_dtype, trans_a, trans_b): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_scatter_add_ext.cc | 30 TypePtr src_dtype; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/ |
| D | conv2d.cc | 324 void ConvertFilter(void *src, void *dst, TypeId src_dtype, TypeId dst_dtype, FilterFormat src_forma… in ConvertFilter() 372 void ConvertFilter(void *src, void *dst, TypeId src_dtype, TypeId dst_dtype, FilterFormat src_forma… in ConvertFilter() 446 auto src_dtype = filter_tensor->data_type(); in InitFilter() local
|
| D | winograd.cc | 166 auto src_dtype = kNumberTypeFloat32; in InitFilter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/ |
| D | insert_quant_node_manager.cc | 341 TypeId src_dtype; in InsertForwardQuantNodeNew() local 425 TypeId src_dtype; in InsertForwardQuantNode() local 518 TypeId src_dtype = kNumberTypeInt8; in InsertBackwardDeQuantNode() local 650 … size_t input_index, TypeId src_dtype, TypeId dst_dtype, in InsertQuantDtypeCastFlyNode() 697 … const AnfNodePtr input_node, TypeId src_dtype, in CreateQuantInputCastNode() 835 … size_t input_index, TypeId src_dtype, TypeId dst_dtype, int axis, in InsertAscendAntiQuantNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | dynamic_gru_v2_grad_fission.cc | 68 auto src_dtype = common::AnfAlgo::GetOutputInferDataType(node, 0); in CastNodeToSpecificDType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/graph_capture/ |
| D | graph_build.cc | 743 auto src_dtype = attr->GetVobj()->GetPyObject().attr("dtype"); in DoMixedPrecisionAttrAccess() local 2679 auto src_dtype = paramNode->GetVobj()->GetPyObject().attr("dtype"); in SetMixedPrecisionType() local
|