| /third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/ |
| D | test_clip_by_norm_no_div_sum.py | 47 def run_clip_by_norm_no_div_sum(shape0, shape1, shape2, shape3, dtype): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | arithmetic_base.cc | 112 auto shape0 = in_tensors_[FIRST_INPUT]->shape(); in ResetStatus() local
143 auto shape0 = a_matric_.shape; in OptimizeShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/sequence/ |
| D | tensor_to_sequence_cpu_kernel.cc | 49 auto shape0 = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_segment_sqrt_n_grad.cc | 51 auto shape0 = inputx->GetTensorShape(); in Compute() local
|
| D | sparse_segment_mean_grad.cc | 56 auto shape0 = ctx.Input(0)->GetTensorShape(); in CheckShapePara() local
|
| D | cholesky_grad.cc | 42 auto shape0 = input0->GetTensorShape(); in Compute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | memcpy_cpu_kernel.cc | 42 auto shape0 = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| D | matmul_cpu_kernel.cc | 293 auto shape0 = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/ |
| D | arithmetic.cc | 200 auto shape0 = in_tensors_.at(0)->shape(); in InitGpuTensorInfoShape() local
|
| D | matmul.cc | 34 auto shape0 = in_tensors_.at(0)->shape(); in IsUseStrassenMatmul() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/ |
| D | combined_non_max_suppression_gpu_kernel.cc | 120 std::vector<int64_t> shape0 = {batch_size_, per_detections_, DimSize4}; in UpdateOutputShapeAndSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/ |
| D | transpose02314_impl.py | 106 def shape0(tik_instance, input_x, res, dtype): function
|
| D | fused_abs_max1_impl.py | 78 def shape0(tik_instance, input_x_shape, input_x, res): function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | common_infer.c | 146 bool ShapeEqual(const int *shape0, size_t shape0_size, const int *shape1, size_t shape1_size) { in ShapeEqual()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | arithmetic.c | 283 int shape0[MAX_LEN] = {0}; in OptimizeArithmeticShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/model/ |
| D | op_node.cc | 560 std::vector<int64_t> shape0 = inputs[0]->shape; in InferShape() local
578 std::vector<int64_t> shape0 = inputs[0]->shape; in InferShape() local
|