Home
last modified time | relevance | path

Searched defs:input_tensor_shape (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dself_adjoint_eig.cc68 auto input_tensor_shape = input_tensor->GetTensorShape(); in SelfAdjointEigCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/common/
Dkernel_base.cc242 aicpuops::TensorShape *input_tensor_shape = input_tensor->mutable_tensor_shape(); in UpdateInputShape() local