Searched defs:input_shape_map (Results 1 – 11 of 11) sorted by relevance
50 auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in UniqueConsecutiveInferShape() local111 auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[0]->GetShape()); in UniqueConsecutiveFrontendInferShape() local
54 …auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->G… in Dropout2DInferShape() local78 …auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->G… in Dropout3DInferShape() local
72 auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_shape_ptr); in InferShape() local
85 …auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->G… in CudnnGRUInferShape() local
54 …auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex0]->G… in LrnInferShape() local
56 auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_shape_ptr); in UCSInferShape() local
135 std::map<int32_t, std::vector<int32_t>> input_shape_map; in ParseInputShapeVector() local
53 auto input_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_shape_ptr); in LrnGradInferShape() local
36 std::map<int32_t, std::vector<int32_t>> input_shape_map; member
212 const std::unordered_map<Tensor *, std::vector<int>> &input_shape_map) { in CheckGraphInputShapes()
470 std::map<int, std::vector<int>> input_shape_map = in Java_com_mindspore_config_MSContext_addDeviceInfo__JLcom_mindspore_config_AscendDeviceInfo_2() local