Searched refs:tpu_target (Results 1 – 9 of 9) sorted by relevance
59 if (options.compile_options.tpu_target == in UpdateTpuTargetByBridgeCompatibility()65 options.compile_options.tpu_target = in UpdateTpuTargetByBridgeCompatibility()68 options.compile_options.tpu_target = in UpdateTpuTargetByBridgeCompatibility()72 LOG(INFO) << "TFRT uses TPU target " << options.compile_options.tpu_target; in UpdateTpuTargetByBridgeCompatibility()
278 tensorflow::TfrtTpuInfraTarget tpu_target) { in CreateResourceContext() argument285 if (tpu_target == tensorflow::TfrtTpuInfraTarget::kTpurt) { in CreateResourceContext()427 runtime(), tpu_model_resource_, options_.compile_options.tpu_target); in ImportAndCompileClientGraph()
76 tensorflow::TfrtTpuInfraTarget tpu_target);
22 std::ostream& operator<<(std::ostream& os, TfrtTpuInfraTarget tpu_target) { in operator <<() argument23 switch (tpu_target) { in operator <<()
33 std::ostream& operator<<(std::ostream& os, TfrtTpuInfraTarget tpu_target);65 TfrtTpuInfraTarget tpu_target = TfrtTpuInfraTarget::kNoTpu; member
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()
174 tensorflow::TfrtTpuInfraTarget tpu_target) { in DefaultTpuModelOptions() argument182 compile_options.tpu_target = tpu_target; in DefaultTpuModelOptions()
113 tensorflow::TfrtTpuInfraTarget tpu_target);
567 options.graph_execution_options.compile_options.tpu_target); in LoadSavedModel()984 options_.graph_execution_options.compile_options.tpu_target); in LoadJoinedSignature()