Searched refs:TfrtTpuInfraTarget (Results 1 – 8 of 8) sorted by relevance
22 std::ostream& operator<<(std::ostream& os, TfrtTpuInfraTarget tpu_target) { in operator <<()24 case TfrtTpuInfraTarget::kNoTpu: in operator <<()26 case TfrtTpuInfraTarget::kTpurt: in operator <<()28 case TfrtTpuInfraTarget::kTfFallback: in operator <<()30 case TfrtTpuInfraTarget::kBridgeFallback: in operator <<()
25 enum class TfrtTpuInfraTarget { enum33 std::ostream& operator<<(std::ostream& os, TfrtTpuInfraTarget tpu_target);65 TfrtTpuInfraTarget tpu_target = TfrtTpuInfraTarget::kNoTpu;
68 if (options.tpu_target == TfrtTpuInfraTarget::kTpurt) { in ConvertTfMlirToBef()93 } else if (options.tpu_target == TfrtTpuInfraTarget::kTfFallback) { in ConvertTfMlirToBef()123 (options.tpu_target == TfrtTpuInfraTarget::kTpurt); in ConvertTfMlirToBef()
60 tensorflow::TfrtTpuInfraTarget::kBridgeFallback) { in UpdateTpuTargetByBridgeCompatibility()66 tensorflow::TfrtTpuInfraTarget::kTfFallback; in UpdateTpuTargetByBridgeCompatibility()69 tensorflow::TfrtTpuInfraTarget::kTpurt; in UpdateTpuTargetByBridgeCompatibility()
76 tensorflow::TfrtTpuInfraTarget tpu_target);
278 tensorflow::TfrtTpuInfraTarget tpu_target) { in CreateResourceContext()285 if (tpu_target == tensorflow::TfrtTpuInfraTarget::kTpurt) { in CreateResourceContext()
113 tensorflow::TfrtTpuInfraTarget tpu_target);
174 tensorflow::TfrtTpuInfraTarget tpu_target) { in DefaultTpuModelOptions()