| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | tril_triu_impl.cu | 25 __global__ void Tril(const size_t size, const T *input, const int diagonal, const int64_t matrix_ro… in Tril()
37 __global__ void Triu(const size_t size, const T *input, const int diagonal, const int64_t matrix_ro… in Triu()
49 __global__ void Triu(const size_t size, const Complex<float> *input, const int diagonal, const int6… in Triu()
64 __global__ void Triu(const size_t size, const Complex<double> *input, const int diagonal, const int… in Triu()
79 cudaError_t CalTril(const size_t size, const T *input, const int diagonal, const int64_t matrix_row, in CalTril()
87 cudaError_t CalTriu(const size_t size, const T *input, const int diagonal, const int64_t matrix_row, in CalTriu()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_tril_triu_op.py | 28 def __init__(self, nptype, diagonal): argument
41 def __init__(self, diagonal=0): argument
51 def __init__(self, diagonal=0): argument
61 def __init__(self, nptype, diagonal): argument
72 def tril_triu(nptype, diagonal): argument
84 def tril_triu_pynative(nptype, diagonal): argument
|
| D | test_func_tril.py | 24 def construct(self, x, diagonal=0): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/ |
| D | test_ops_triu.cc | 34 ValuePtr diagonal; member
44 auto diagonal = param.diagonal->ToAbstract(); in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/scipy/ |
| D | ops_wrapper.py | 23 def matrix_set_diag(input_x, diagonal, k=0, alignment="RIGHT_LEFT"): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | tril.cc | 64 void Tril::Init(const int64_t diagonal) { set_diagonal(diagonal); } in Init()
66 void Tril::set_diagonal(const int64_t diagonal) { (void)this->AddAttr(kDiagonal, api::MakeValue(dia… in set_diagonal()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/dynamic_shape/grad/ |
| D | test_matrix_set_diag_v3.py | 30 def construct(self, x, diagonal, k): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/ |
| D | triu_mapper.cc | 72 auto diagonal = diagonal_data[0]; in Mapper() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_matrix_set_diag_v3_op.py | 31 def construct(self, x, diagonal, k): argument
|
| D | test_svd_op.py | 37 def matrix_diag(diagonal, shape): argument
|
| D | test_matrix_diag_v3_op.py | 48 def benchmark(diagonal, expect, align="RIGHT_LEFT", k=None, num_rows=None, num_cols=None, padding_v… argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/tensor/ |
| D | test_tril.py | 23 def construct(self, x, diagonal=0): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_svd_op.py | 37 def matrix_diag(diagonal, shape): argument
|
| D | test_matrix_diag_v3_op.py | 35 def benchmark(diagonal, expect, align="RIGHT_LEFT", k=None, num_rows=None, num_cols=None, padding_v… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | tril.cc | 73 AttrValue *diagonal = ctx.GetAttr("diagonal"); in ValidParam() local
|
| /third_party/skia/m133/bench/ |
| D | ShapesBench.cpp | 108 float diagonal = sqrtf(static_cast<float>(fShapesSize.width() * fShapesSize.width()) + in clampShapeSize() local
|
| /third_party/skia/bench/ |
| D | ShapesBench.cpp | 106 float diagonal = sqrtf(static_cast<float>(fShapesSize.width() * fShapesSize.width()) + in clampShapeSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/expander/core/ |
| D | emitter.h | 170 …NodePtr MatrixSetDiagV3(const NodePtr &x, const NodePtr &diagonal, const NodePtr &k, const ValuePt… in MatrixSetDiagV3()
175 …NodePtr MatrixDiagPartV3(const NodePtr &x, const NodePtr &diagonal, const NodePtr &k, const ValueP… in MatrixDiagPartV3()
|
| /third_party/mesa3d/src/vulkan/runtime/bvh/ |
| D | vk_build_helpers.h | 265 vec3 diagonal = aabb.max - aabb.min; in aabb_surface_area() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | basic.py | 1378 def construct(self, input_x, diagonal): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_grad_experimental/ |
| D | grad_array_ops.py | 258 def bprop(x, diagonal, k, out, dout): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/numpy/ |
| D | array_creations.py | 1324 def diagonal(a, offset=0, axis1=0, axis2=1): function
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | array_func.py | 3874 def matrix_set_diag(x, diagonal, k=0, align="RIGHT_LEFT"): # pylint: disable=redefined-outer-name argument
4916 def tril(input, diagonal=0): # pylint: disable=redefined-outer-name argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 366 def tril(x, diagonal=0): argument
1383 def diagonal(x, offset=0, axis1=0, axis2=1): function
3423 def triu(input, diagonal=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 1317 def triu(self, diagonal=0): argument
2154 def tril(self, diagonal=0): argument
2951 def diagonal(self, offset=0, axis1=0, axis2=1): member in Tensor
|