| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | fused_abs_max1_impl.py | 76 def shape0(tik_instance, input_x_shape, input_x, res): argument
100 def shape1(tik_instance, input_x_shape, ori_shape, input_x, res): argument
146 def shape2(tik_instance, input_x_shape, input_x, res): argument
242 def shape3(tik_instance, input_x_shape, ori_shape, input_x, res): argument
275 def shape4(tik_instance, input_x_shape, input_x, res): argument
302 def shape5(tik_instance, input_x_shape, input_x, res): argument
362 def shape7(tik_instance, input_x_shape, input_x, res): argument
392 def shape8(tik_instance, input_x_shape, input_x, res): argument
424 def shape9(tik_instance, input_x_shape, input_x, res): argument
465 def shape10(tik_instance, input_x_shape, input_x, res): argument
|
| D | transpose02314_impl.py | 42 def _error_feedback(input_x_shape): argument
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | reduce.cc | 36 std::vector<int64_t> infer_shape_reduce(std::vector<int64_t> input_x_shape, const ValuePtr axis_val… in infer_shape_reduce()
73 …auto input_x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->BuildShape())[… in InferShape() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _embedding_cache_ops.py | 50 def check_shape(self, input_x_shape, indices_shape, update_shape, max_num_shape): argument
95 def check_shape(self, input_x_shape, max_num_shape, offset_shape): argument
302 def check_shape(self, input_x_shape, cum_sum_arr_shape, shift_idx_shape): argument
|
| D | array_ops.py | 3824 def infer_shape(self, input_x_shape, indices_shape, updates_shape): argument
3909 def infer_shape(self, input_x_shape, indices_shape, updates_shape): argument
6236 def infer_shape(self, input_x_shape, indices_shape, updates_shape): argument
6320 def infer_shape(self, input_x_shape, indices_shape, updates_shape): argument
6405 def infer_shape(self, input_x_shape, indices_shape, updates_shape): argument
|
| D | _quant_ops.py | 226 def infer_shape(self, input_x_shape, alpha_shape, quant_max_shape): argument
383 def infer_shape(self, input_x_shape, alpha_shape, quant_max_shape): argument
|
| D | nn_ops.py | 3861 def infer_shape(self, input_x_shape, weight_shape): argument
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | soft_shrink_grad.cc | 39 …auto input_x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[… in SoftShrinkGradInferShape() local
|
| D | hsigmoid_grad.cc | 43 …auto input_x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[1]->BuildShape())[… in InferShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | map_uniform_cpu_kernel.cc | 58 auto input_x_shape = AnfAlgo::GetPrevNodeOutputInferShape(node, 0); in LaunchKernel() local
|
| D | dynamic_assign_cpu_kernel.cc | 65 auto input_x_shape = AnfAlgo::GetPrevNodeOutputInferShape(node, 0); in LaunchKernel() local
|
| D | assign_cpu_kernel.cc | 41 auto input_x_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); in InitKernel() local
|
| D | l2normalize_grad_cpu_kernel.cc | 79 auto input_x_shape = input_shape_list_[0]; in CheckInputShape() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.cc | 260 auto input_x_shape = AnfAlgo::GetOutputInferShape(anf_node, 0); in GenTopKV2IndicesTensorInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1069 auto input_x_shape = dyn_cast<abstract::Shape>(node->input(kOneNum)->Shape()); in ExportPrimBatchMatMul() local
|