@@ -404,6 +404,8 @@ TfLiteAllocationStrategy TfLiteTensorGetAllocationStrategy(
404404 return kTfLiteAllocationStrategyUnknown ;
405405 case kTfLiteVariantObject :
406406 return kTfLiteAllocationStrategyNew ;
407+ case kTfLiteNonCpu :
408+ return kTfLiteAllocationStrategyUnknown ;
407409 }
408410 return kTfLiteAllocationStrategyUnknown ;
409411}
@@ -428,6 +430,8 @@ TfLiteRunStability TfLiteTensorGetBufferAddressStability(
428430 return kTfLiteRunStabilityUnknown ;
429431 case kTfLiteVariantObject :
430432 return kTfLiteRunStabilityAcrossRuns ;
433+ case kTfLiteNonCpu :
434+ return kTfLiteRunStabilityUnknown ;
431435 }
432436 return kTfLiteRunStabilityUnknown ;
433437}
@@ -451,6 +455,8 @@ TfLiteRunStability TfLiteTensorGetDataStability(const TfLiteTensor* const t) {
451455 return kTfLiteRunStabilityUnknown ;
452456 case kTfLiteVariantObject :
453457 return kTfLiteRunStabilitySingleRun ;
458+ case kTfLiteNonCpu :
459+ return kTfLiteRunStabilityUnknown ;
454460 }
455461 return kTfLiteRunStabilityUnknown ;
456462}
@@ -477,11 +483,13 @@ TfLiteRunStep TfLiteTensorGetDataKnownStep(const TfLiteTensor* t) {
477483 return kTfLiteRunStepUnknown ;
478484 case kTfLiteVariantObject :
479485 return kTfLiteRunStepEval ;
486+ case kTfLiteNonCpu :
487+ return kTfLiteRunStepUnknown ;
480488 }
481489 return kTfLiteRunStepUnknown ;
482490}
483491
484- // Returns the operation steop when the shape of a tensor is computed.
492+ // Returns the operation step when the shape of a tensor is computed.
485493//
486494// Some operations can precompute the shape of their results before the
487495// evaluation step. This makes the shape available earlier for subsequent
@@ -504,6 +512,8 @@ TfLiteRunStep TfLiteTensorGetShapeKnownStep(const TfLiteTensor* t) {
504512 return kTfLiteRunStepUnknown ;
505513 case kTfLiteVariantObject :
506514 return kTfLiteRunStepEval ;
515+ case kTfLiteNonCpu :
516+ return kTfLiteRunStepUnknown ;
507517 }
508518 return kTfLiteRunStepUnknown ;
509519}
0 commit comments