Searched refs:KernelFallbackOutputType (Results 1 – 3 of 3) sorted by relevance
57 const tfrt::OpAttrsRef& attrs, KernelFallbackOutputType output_type) { in KernelFallbackExecute()111 if (output_type == KernelFallbackOutputType::TENSOR) { in KernelFallbackExecute()116 KernelFallbackOutputType::KERNEL_FALLBACK_TENSOR); in KernelFallbackExecute()
35 enum KernelFallbackOutputType { enum47 const tfrt::OpAttrsRef& attrs, KernelFallbackOutputType output_type);
65 tfd::KernelFallbackOutputType::TENSOR); in TFDForwardKernel()