Home
last modified time | relevance | path

Searched defs:GetTensorShape (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/
Dhccl_kernel.cc196 std::function<ShapeVector(KernelTensor *)> GetTensorShape; in CalcTypeShapeAndCount() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dfunc_graph_utils.cc238 ShapeVector FuncGraphUtils::GetTensorShape(const AnfWithOutIndex &tensor) { in GetTensorShape() function in mindspore::FuncGraphUtils
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dflash_attention_fusion.cc88 std::vector<int64_t> GetTensorShape(CNodePtr cnode, size_t input_index) { in GetTensorShape() function
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dgroupnormsilu_fusion.cc36 std::vector<int64_t> GetTensorShape(CNodePtr cnode, size_t input_index) { in GetTensorShape() function
Dflash_attention_fusion.cc294 std::vector<int64_t> GetTensorShape(CNodePtr cnode, size_t input_index) { in GetTensorShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Ddct_cpu_kernel.cc64 inline Eigen::array<Eigen::DenseIndex, x_rank> GetTensorShape(const std::vector<int64_t> &x_shape) { in GetTensorShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Ddct.cc152 inline Eigen::array<Eigen::DenseIndex, x_rank> GetTensorShape(const std::vector<int64_t> &x_shape) { in GetTensorShape() function