Searched defs:indices_dtype (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 69 def infer_dtype(self, x_dtype, indices_dtype, updates_dtype): argument
109 def check_dtype(self, x_dtype, indices_dtype, updates_dtype): argument
3767 def infer_dtype(self, x_dtype, indices_dtype): argument
3845 def infer_dtype(self, input_x_dtype, indices_dtype, updates_dtype): argument
3930 def infer_dtype(self, input_x_dtype, indices_dtype, updates_dtype): argument
4055 def infer_dtype(self, x_dtype, indices_dtype, value_dtype): argument
4795 def infer_dtype(self, x_dtype, indices_dtype, updates_dtype): argument
6257 def infer_dtype(self, input_x_dtype, indices_dtype, updates_dtype): argument
6341 def infer_dtype(self, input_x_dtype, indices_dtype, updates_dtype): argument
6426 def infer_dtype(self, input_x_dtype, indices_dtype, updates_dtype): argument
|
| D | _embedding_cache_ops.py | 53 def check_dtype(self, input_x_dtype, indices_dtype, update_dtype, max_num_dtype): argument
|
| D | nn_ops.py | 4968 beta1_dtype, beta2_dtype, epsilon_dtype, grad_dtype, indices_dtype): argument
5119 beta1_dtype, beta2_dtype, epsilon_dtype, grad_dtype, indices_dtype): argument
5232 def infer_dtype(self, var_dtype, accum_dtype, linear_dtype, grad_dtype, indices_dtype): argument
5335 grad_dtype, indices_dtype): argument
6412 grad_dtype, indices_dtype): argument
7132 def check_dtype(self, var_dtype, accum_dtype, linear_dtype, grad_dtype, indices_dtype): argument
7241 def infer_dtype(self, var_dtype, accum_dtype, linear_dtype, grad_dtype, indices_dtype): argument
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_others.cc | 212 auto indices_dtype = indices->element()->BuildType(); in InferImplMakeRowTensor() local
318 auto indices_dtype = indices->element()->BuildType(); in InferImplMakeSparseTensor() local
|