Home
last modified time | relevance | path

Searched defs:matrix_tensor (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dshape_fusion_infer.c20 int CalculateOutput(const TensorC *in_tensor, const TensorC *matrix_tensor, TensorC *out_tensor, si… in CalculateOutput()
81 const TensorC *matrix_tensor = inputs[out_idx + 1]; in ShapeFusionInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dmatrix_triangular_solve.cc105 Tensor *matrix_tensor = ctx.Input(0); in MatrixTriangularSolveCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dlinalg_ops_proto.cc28 auto matrix_tensor = op.get_input_desc_matrix(); in IMPLEMT_INFERFUNC() local
86 auto matrix_tensor = op.get_input_desc_matrix(); in IMPLEMT_INFERFUNC() local