| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | fse_encoder.cc | 29 constexpr int kInt16 = 16; variable
80 delta_bit_count[sym] = (max_bits_out << kInt16) - min_state_plus; in FSECreateStatesForEncoding()
85 delta_bit_count[sym] = (table_log << kInt16) - (1 << table_log); in FSECreateStatesForEncoding()
162 ret = bs.Create(kInt16 * fse_quant.symbol_table_count); in Compress()
197 uint8_t bits_out = (state + delta_bit_count[sym]) >> kInt16; in FSEEncodeSymbolGetNewState()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | sigmoid_cross_entropy_with_logits.cc | 43 …const std::set<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUIn… in SigmoidCrossEntropyWithLogitsInfer()
|
| D | hsigmoid.cc | 38 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}; in InferType()
|
| D | reverse_v2.cc | 35 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, kUInt16, in InferType()
|
| D | zeros.cc | 48 const std::set<TypePtr> valid_types = {kBool, kInt8, kInt16, kInt32, kInt64, kUInt8, in ZerosInferType()
|
| D | ones.cc | 46 const std::set<TypePtr> valid_types = {kBool, kInt8, kInt16, kInt32, kInt64, kUInt8, in OnesInferType()
|
| D | op_utils.h | 281 const std::set<TypePtr> common_valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, kUInt1…
285 kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUInt, kUInt8,
|
| D | gather_nd.cc | 53 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64}; in InferType()
|
| D | embedding_lookup.cc | 49 const std::set<TypePtr> int_valid_types = {kInt8, kInt16, kInt32, kInt64}; in EmbeddingLookupInfer()
|
| D | index_add.cc | 66 …const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kUInt8, kFloat16, kFloat32, kFloat64… in IndexAddInferType()
|
| D | one_hot.cc | 80 {kInt8, kInt16, kInt32, kInt64}, op_name); in OneHotInferType()
|
| D | split_v.cc | 88 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kUInt8, in InferType()
|
| D | gather.cc | 42 std::set<TypePtr> int_types = {kInt8, kInt16, kInt32, kInt64}; in GatherInfer()
|
| D | mat_mul.cc | 95 …const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32, kFloat64… in MatMulInferType()
|
| D | batch_matmul.cc | 121 …const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32, kFloat64… in BatchMatmulInferType()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | sigmoid_cross_entropy_with_logits_grad.cc | 46 …const std::set<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUIn… in SigmoidCrossEntropyWithLogitsGradInfer()
|
| D | hsigmoid_grad.cc | 52 const std::set<TypePtr> valid_types = {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}; in InferType()
|
| D | smooth_l1_loss_grad.cc | 53 …const std::set<TypePtr> valid_types = {kBool, kInt, kInt8, kInt16, kInt32, kInt64, kUIn… in SmoothL1LossGradInfer()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | scalar.h | 120 Int16Imm() : IntergerImm(kInt16), v_(0) {} in IMM_TRAITS()
121 …explicit Int16Imm(int16_t v) : IntergerImm(kInt16), v_(v) { hash_ = hash_combine({tid(), std::hash… in IMM_TRAITS()
|
| D | dtype_extends.cc | 67 …{kNumberTypeInt16, kInt16}, {kNumberTypeInt32, kInt32}, {kNumberTypeInt, kInt32}, in TypeIdToType()
|
| /third_party/mindspore/mindspore/core/ir/dtype/ |
| D | number.h | 170 inline const TypePtr kInt16 = std::make_shared<Int>(static_cast<int>(BitsNum::eBits16)); variable
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 405 …(void)CheckTensorDType(tensor, {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}, "Input of HSig… in InferImplHSigmoid()
416 (void)CheckTensorDType(dout, {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}, in InferImplHSigmoidGrad()
418 …(void)CheckTensorDType(x, {kInt8, kInt16, kInt32, kInt64, kFloat16, kFloat32}, "X of HSigmoidGrad … in InferImplHSigmoidGrad()
|
| /third_party/mindspore/tests/ut/cpp/ir/ |
| D | meta_tensor_test.cc | 212 TensorPtr tensor_int16 = TensorPy::MakeTensor(py::array(tuple), kInt16); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 783 } else if (type_id == kInt16->type_id()) { in GetInputTensorValue()
|