Searched refs:GetInferenceDeviceTypeForOp (Results 1 – 3 of 3) sorted by relevance
329 auto inference_device_type = GetInferenceDeviceTypeForOp(impl); in BuildSubgraphs()430 GetInferenceDeviceTypeForOp(impl); in RewireSubgraphs()
347 auto current_device_type = GetInferenceDeviceTypeForOp(&op); in RaiseTargetSubgraphsForBlock()
129 inline llvm::Optional<InferenceDeviceType> GetInferenceDeviceTypeForOp( in GetInferenceDeviceTypeForOp() function