| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_matrix_diag_v3_op.py | 31 def construct(self, x, k, num_rows, num_cols, padding_value): argument
42 def construct(self, x, k, num_rows, num_cols, padding_value): argument
48 …ark(diagonal, expect, align="RIGHT_LEFT", k=None, num_rows=None, num_cols=None, padding_value=None, argument
|
| D | test_matrix_diag_part_v3_op.py | 29 def construct(self, x, k, padding_value): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_matrix_diag_v3_op.py | 30 def construct(self, x, k, num_rows, num_cols, padding_value): argument
35 …ark(diagonal, expect, align="RIGHT_LEFT", k=None, num_rows=None, num_cols=None, padding_value=None, argument
|
| D | test_matrix_diag_part_v3_op.py | 29 def construct(self, x, k, padding_value): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/ |
| D | test_matrix_diag_partv3.py | 31 def construct(self, x, k, padding_value): argument
|
| D | test_matrix_diag_v3.py | 30 def construct(self, x, k, num_rows, num_cols, padding_value): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | matrix_diag_v3_impl.cu | 36 DataType padding_value = *padding_value_ptr; in MatrixDiagV3Kernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | matrix_diag_part_v3.cc | 119 …auto padding_value = CheckAndConvertUtils::CheckArgsType(prim_name, input_args, kInputIndex2, kObj… in MatrixDiagPartV3InferType() local
|
| D | matrix_diag_v3.cc | 187 …auto padding_value = CheckAndConvertUtils::CheckArgsType(prim_name, input_args, kInputIndex4, kObj… in MatrixDiagV3InferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | gather_v2_ds_fission.cc | 94 auto padding_value = MakeValue(padding_vector); in CreatePad() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | matrix_diag_v3_cpu_kernel.cc | 207 T padding_value = padding_value_data[0]; in DoLaunch() local
|
| D | matrix_diag_part_v3_cpu_kernel.cc | 161 T padding_value = padding_value_data[0]; in DoLaunch() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/expander/fallback/ |
| D | nn_ops.cc | 717 auto padding_value = paddings->BuildValue(); in PaddingTupleToTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | array_func.py | 3709 def matrix_diag(x, k=0, num_rows=-1, num_cols=-1, padding_value=0, align="RIGHT_LEFT"): argument
3810 def matrix_diag_part(x, k, padding_value, align="RIGHT_LEFT"): argument
4710 def col2im(input_x, output_size, kernel_size, dilation, padding_value, stride): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/ops/ |
| D | test_ops.py | 1337 def __init__(self, k, num_rows, num_cols, padding_value, align='LEFT_RIGHT'): argument
1345 def construct(self, x, k, num_rows, num_cols, padding_value): argument
1350 def __init__(self, k, padding_value, align='LEFT_RIGHT'): argument
1356 def construct(self, x, k, padding_value): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_grad_experimental/ |
| D | grad_array_ops.py | 225 def bprop(x, k, padding_value, out, dout): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 1923 def col2im(self, output_size, kernel_size, dilation, padding_value, stride): argument
|