| /third_party/mindspore/mindspore/ops/operations/ |
| D | comm_ops.py | 164 def infer_dtype(self, x_dtype): argument
241 def infer_dtype(self, x_dtype): argument
278 def infer_dtype(self, x_dtype, z_shape): argument
309 def infer_dtype(self, x_dtype, z_dtype): argument
359 def infer_dtype(self, x_dtype): argument
437 def infer_dtype(self, x_dtype): argument
490 def infer_dtype(self, x_dtype): argument
564 def infer_dtype(self, x_dtype): argument
705 def infer_dtype(self, x_dtype): argument
735 def infer_dtype(self, x_dtype): argument
[all …]
|
| D | math_ops.py | 93 def do_infer_dtype(x_dtype, y_dtype, valid_dtype=mstype.number_type, prim_name=None): argument
116 def infer_dtype(self, x_dtype, y_dtype): argument
1516 def infer_dtype(self, x_dtype): argument
1577 def infer_dtype(self, x_dtype, v_dtype): argument
1646 def infer_dtype(self, x_dtype, v_dtype): argument
1808 def infer_dtype(self, x_dtype, y_dtype): argument
1886 def infer_dtype(self, x_dtype): argument
2198 def infer_dtype(self, x_dtype, range_dtype): argument
2328 def infer_dtype(self, x_dtype): argument
2936 def infer_dtype(self, x_dtype): argument
[all …]
|
| 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
686 def infer_dtype(self, x_dtype): argument
1686 def infer_dtype(self, x_dtype): argument
1743 def infer_dtype(self, x_dtype): argument
1811 def infer_dtype(self, x_dtype): argument
1879 def infer_dtype(self, x_dtype): argument
2883 def infer_dtype(self, x_dtype): argument
2927 def infer_dtype(self, x_dtype): argument
3713 def infer_dtype(self, x_dtype): argument
[all …]
|
| D | _inner_ops.py | 170 def infer_dtype(self, x_dtype): argument
314 def infer_dtype(self, x_dtype, assist_dtype): argument
363 def infer_dtype(self, x_dtype, assist_dtype): argument
430 def infer_dtype(self, x_dtype): argument
491 def infer_dtype(self, x_dtype=None): argument
522 def infer_dtype(self, x_dtype, diagonal_dtype, assist_dtype): argument
|
| D | _grad_ops.py | 131 def infer_dtype(self, x_dtype, dout_dtype): argument
148 def infer_dtype(self, x_dtype, dout_dtype): argument
171 def infer_dtype(self, x_dtype, dout_dtype): argument
733 def infer_dtype(self, y_backprop_dtype, x_dtype, y_dtype): argument
751 def infer_dtype(self, y_backprop_dtype, x_dtype): argument
997 def infer_dtype(self, x_dtype, y_dtype, grad_dtype): argument
1028 def infer_dtype(self, x_dtype, y_dtype, grad_dtype): argument
1059 def infer_dtype(self, x_dtype, grad_dtype, argmax_dtype): argument
1105 def infer_dtype(self, x_dtype, grad_dtype, argmax_dtype): argument
1308 def infer_dtype(self, x_dtype, hx_dtype, y_dtype, reserve_dtype, state_dtype): argument
[all …]
|
| D | nn_ops.py | 233 def infer_dtype(self, x_dtype): argument
502 def infer_dtype(self, x_dtype): argument
556 def infer_dtype(self, x_dtype): argument
753 def infer_dtype(self, x_dtype): argument
1605 def infer_dtype(self, x_dtype, w_dtype, b_dtype=None): argument
1684 def infer_dtype(self, x_dtype): argument
1829 def infer_dtype(self, x_dtype): argument
1984 def infer_dtype(self, x_dtype): argument
2468 def infer_dtype(self, x_dtype, t_dtype, w_dtype): argument
2889 def infer_dtype(self, x_dtype): argument
[all …]
|
| D | _thor_ops.py | 528 def infer_dtype(self, x_dtype): argument
602 def infer_dtype(self, x_dtype): argument
|
| D | random_ops.py | 435 def infer_dtype(self, x_dtype): argument
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | reluv2.cc | 30 …GetOutputMaskShape(const std::vector<int64_t> &input_shape, const std::shared_ptr<Type> &x_dtype) { in GetOutputMaskShape()
71 auto x_dtype = input_type->element(); in InferShape() local
|
| D | cos.cc | 32 auto x_dtype = input_args[0]->BuildType(); in CosInferType() local
|
| D | diag.cc | 39 auto x_dtype = input_args[0]->BuildType(); in PartInferType() local
|
| D | diag_part.cc | 49 auto x_dtype = input_args[0]->BuildType(); in DiagPartInferType() local
|
| D | arg_min.cc | 60 auto x_dtype = input_args[0]->BuildType()->cast<TensorTypePtr>()->element(); in ArgMinInfer() local
|
| D | broadcast_to.cc | 69 auto x_dtype = input_args[0]->BuildType()->cast<TensorTypePtr>(); in BroadcastToInferType() local
|
| D | square.cc | 45 auto x_dtype = input_args[kInputIndex0]->BuildType(); in SquareInferType() local
|
| D | tile.cc | 87 auto x_dtype = x_type_map->cast<TensorTypePtr>(); in TileInferType() local
|
| D | avg_pool_3d.cc | 166 auto x_dtype = input_args[0]->BuildType(); in InferType() local
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_inner_ops.py | 26 def _get_matrix_diag_assist(x_shape, x_dtype): argument
33 def _get_matrix_diag_part_assist(x_shape, x_dtype): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | avg_pool_grad_unify_mindir.cc | 102 const PadMode pad_mode, const TypeId x_dtype) { in CreateMeanMatrixValueNode()
157 const std::vector<int64_t> &k_size, const TypeId x_dtype) { in CreateKernelMatrixValueNode()
194 auto x_dtype = AnfAlgo::GetPrevNodeOutputInferDataType(avgpool_grad, 0); in Process() local
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | basic.py | 351 def _dtype_check(x_dtype, prim_name=None): argument
999 def tril(x_shape, x_dtype, k): argument
1086 def triu(x_shape, x_dtype, k): argument
1173 def _get_matrix_diag_assist(x_shape, x_dtype): argument
1181 def _get_matrix_diag_part_assist(x_shape, x_dtype): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | space_to_depth_split.cc | 108 TypeId x_dtype = AnfAlgo::GetOutputInferDataType(ori_inputs[kIndex1], 0); in Process() local
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_shift_op.py | 48 def infer_dtype(self, x_dtype, fill_value_type): argument
|
| D | test_rolling_op.py | 56 def infer_dtype(self, x_dtype): argument
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_math_ops.py | 156 def infer_dtype(self, x_dtype, out_dtype, dout_dtype): argument
182 def infer_dtype(self, x_dtype): argument
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_debug_location.py | 59 def infer_dtype(self, x_dtype, y_dtype): argument
|