| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/ |
| D | fused_abs_max1_impl.py | 78 def shape0(tik_instance, input_x_shape, input_x, res): argument
102 def shape1(tik_instance, input_x_shape, ori_shape, input_x, res): argument
148 def shape2(tik_instance, input_x_shape, input_x, res): argument
244 def shape3(tik_instance, input_x_shape, ori_shape, input_x, res): argument
277 def shape4(tik_instance, input_x_shape, input_x, res): argument
304 def shape5(tik_instance, input_x_shape, input_x, res): argument
364 def shape7(tik_instance, input_x_shape, input_x, res): argument
394 def shape8(tik_instance, input_x_shape, input_x, res): argument
426 def shape9(tik_instance, input_x_shape, input_x, res): argument
467 def shape10(tik_instance, input_x_shape, input_x, res): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | _embedding_cache_ops.py | 52 def check_shape(self, input_x_shape, indices_shape, update_shape, max_num_shape): argument
97 def check_shape(self, input_x_shape, max_num_shape, offset_shape): argument
303 def check_shape(self, input_x_shape, cum_sum_arr_shape, shift_idx_shape): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | flatten.cc | 27 const auto &input_x_shape = input_args[kIndex0]->GetShape(); in InferShape() local
|
| D | gather_nd.cc | 26 const auto &input_x_shape = input_args[0]->GetShape()->GetShapeVector(); in InferShape() local
|
| D | grid_sampler_3d.cc | 34 auto input_x_shape = input_x_base_shape->GetShapeVector(); in InferShape() local
|
| D | grid_sampler_2d.cc | 36 auto input_x_shape = input_x_base_shape->GetShapeVector(); in InferShape() local
|
| D | fft_arithmetic.cc | 133 const auto &input_x_shape = input_args[kIndex0]->GetShape(); in FFTCheckValidation() local
170 const auto &input_x_shape = input_args[kIndex0]->GetShape(); in FFTNCheckValidation() local
|
| D | grid_sampler_2d_grad.cc | 39 auto input_x_shape = input_x_base_shape->GetShapeVector(); in InferShape() local
|
| D | grid_sampler_3d_grad.cc | 41 auto input_x_shape = input_x_base_shape->GetShapeVector(); in InferShape() local
|
| D | slice_ext.cc | 27 auto input_x_shape = input_args[0]->GetShape()->GetShapeVector(); in InferShape() local
|
| D | irfft_grad.cc | 55 const auto &input_x_shape = input_args[kIndex0]->GetShape(); in CheckValidation() local
|
| D | rfft_grad.cc | 55 const auto &input_x_shape = input_args[kIndex0]->GetShape(); in CheckValidation() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | convert_pad_v3_paddings.cc | 183 auto input_x_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(pad_node, kIndex0); in CreateDynPaddingsPass() local
286 auto input_x_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(pad_node, kIndex0); in CreateConstPaddingsNode() local
301 auto input_x_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(node, kIndex0); in Process() local
342 auto input_x_shape = common::AnfAlgo::GetPrevNodeOutputInferShape(node, kIndex0); in ExpandInputXDims() local
|
| D | avg_pool_grad_for_ge.cc | 45 auto input_x_shape = input_x->Shape(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_expand_dims.cc | 32 ShapeVector input_x_shape; member
|
| D | test_ops_gather_nd.cc | 24 ShapeVector input_x_shape; member
|
| D | test_ops_diag.cc | 32 ShapeVector input_x_shape; member
|
| D | test_ops_bias_add.cc | 33 ShapeVector input_x_shape; member
|
| D | test_ops_hshrink.cc | 32 ShapeVector input_x_shape; member
|
| /third_party/mindspore/mindspore-src/source/tests/st/pi_jit/dynamic_shape_operations/ |
| D | median.py | 28 def __init__(self, input_x_shape, global_median, axis, keep_dims, dtype, loss=0.0001): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | dynamic_assign_cpu_kernel.cc | 57 const auto &input_x_shape = inputs[kIndex0]->GetShapeVector(); in LaunchKernel() local
|
| D | map_uniform_cpu_kernel.cc | 53 auto input_x_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| D | sub_and_filter_cpu_kernel.cc | 45 auto input_x_shape = inputs.at(kIndex0)->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | l2_normalize_grad.cc | 71 auto input_x_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_x_shape_ptr)[kShape]; in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | layer_norm_grad_grad_gpu_kernel.cc | 118 std::vector<int64_t> input_x_shape = inputs[kIndex0]->GetShapeVector(); in Resize() local
|