Searched defs:indices_type (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | scatter_nd.cc | 45 auto indices_type = input_args[0]->BuildType(); in InferType() local
|
| D | scatter_nd_update.cc | 43 auto indices_type = input_args[kInputIndex1]->BuildType(); in InferType() local
|
| D | index_add.cc | 69 auto indices_type = input_args[kInputIndex1]->BuildType(); in IndexAddInferType() local
|
| D | sparse_apply_r_m_s_prop.cc | 84 auto indices_type = input_args[5]->BuildType(); in InferType() local
|
| /third_party/boost/boost/signals2/detail/ |
| D | variadic_slot_invoker.hpp | 89 typedef typename make_unsigned_meta_array<N>::type indices_type; in operator ()() typedef
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _inner_ops.py | 1110 def check_dtype(self, indices_type, data_type): argument
|
| D | nn_ops.py | 6081 def infer_dtype(self, var_type, accum_type, grad_type, indices_type): argument
6184 def infer_dtype(self, var_type, accum_type, grad_type, indices_type): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 283 auto indices_type = indices->Type(); in CreateMapCacheIdx() local
|