Searched defs:GetInputShape (Results 1 – 10 of 10) sorted by relevance
187 const std::vector<ShapeVector> ModelInfer::GetInputShape() { return model_process_.GetInputShape();… in GetInputShape() function in mindspore::kernel::acl::ModelInfer
279 const std::vector<ShapeVector> ModelProcess::GetInputShape() { in GetInputShape() function in mindspore::kernel::acl::ModelProcess
165 std::map<std::string, std::vector<int64_t>> Converter::GetInputShape() const { in GetInputShape() function
49 std::string GetInputShape() const { return input_shape_; } in GetInputShape() function
79 ShapeVector CallbackImpl::GetInputShape(const AnfNodePtr &node, size_t i) { in GetInputShape() function in mindspore::graphkernel::CallbackImpl370 ShapeVector CallbackImplWithInferShape::GetInputShape(const AnfNodePtr &node, size_t i) { in GetInputShape() function in mindspore::graphkernel::CallbackImplWithInferShape
572 std::string AscendDeviceInfo::GetInputShape() const { return CharToString(GetInputShapeChar()); } in GetInputShape() function
32 ShapeVector CallbackImpl::GetInputShape(const AnfNodePtr &node, size_t i) { return GetInputInferSha… in GetInputShape() function in mindspore::graphkernel::CallbackImpl
70 SymbolPtr GetInputShape(size_t i) const { return GetShape(GetInput(i)); } in GetInputShape() function
502 std::vector<int> NodeInferShape::GetInputShape(const CNodePtr &cnode, size_t index) { in GetInputShape() function in mindspore::opt::NodeInferShape
85 std::vector<int64_t> ReshapeInfo::GetInputShape(const AnfNodePtr &shape_input_node) { in GetInputShape() function in mindspore::parallel::ReshapeInfo