Searched defs:output_infer_shape (Results 1 – 4 of 4) sorted by relevance
62 std::vector<ShapeVector> output_infer_shape; in CacheForExecutionOrder() local107 std::vector<ShapeVector> output_infer_shape; in CacheForGraphInputs() local152 const ShapeVector &OpRuntimeInfo::output_infer_shape(size_t index) const { in output_infer_shape() function in mindspore::runtime::OpRuntimeInfo
88 std::vector<size_t> output_tensor_size, std::vector<ShapeVector> output_infer_shape, in OpRuntimeInfo()
33 std::vector<ShapeVector> output_infer_shape; in GetNodeOutputInfo() local55 std::vector<ShapeVector> output_infer_shape; in GetNodesOutputInfo() local77 const auto &output_infer_shape = std::get<kIndex1>(output_info); in GetInputNodeInfoForOneRank() local
38 std::vector<ShapeVector> output_infer_shape; in GetNodeOutputInfo() local