Path Lines of Code backends/gpu/cpp_tests/include/common.h 30 backends/gpu/cpp_tests/instantiate_suite.cc 14 backends/gpu/include/tfrt/gpu/core_runtime/gpu_dispatch_context.h 53 backends/gpu/include/tfrt/gpu/core_runtime/gpu_op_handler.h 14 backends/gpu/include/tfrt/gpu/core_runtime/gpu_op_registry.h 57 backends/gpu/include/tfrt/gpu/core_runtime/gpu_op_utils.h 18 backends/gpu/include/tfrt/gpu/device/conversion_function.h 25 backends/gpu/include/tfrt/gpu/device/device.h 34 backends/gpu/include/tfrt/gpu/device/device_util.h 13 backends/gpu/include/tfrt/gpu/device/gpu_config.h 18 backends/gpu/include/tfrt/gpu/gpu_executor.h 62 backends/gpu/include/tfrt/gpu/gpu_types.h 270 backends/gpu/include/tfrt/gpu/kernels/gpu_ops.h 91 backends/gpu/include/tfrt/gpu/kernels/kernels_detail.h 51 backends/gpu/include/tfrt/gpu/memory/bfc_gpu_allocator.h 67 backends/gpu/include/tfrt/gpu/ops/tf/bias_add_f16_kernel.h 412 backends/gpu/include/tfrt/gpu/ops/tf/bias_add_f32_kernel.h 411 backends/gpu/include/tfrt/gpu/ops/tf/bias_add_f64_kernel.h 412 backends/gpu/include/tfrt/gpu/ops/tf/gpu_ops.h 10 backends/gpu/include/tfrt/gpu/ops/tf/relu_f16_kernel.h 300 backends/gpu/include/tfrt/gpu/ops/tf/relu_f32_kernel.h 296 backends/gpu/include/tfrt/gpu/ops/tf/relu_f64_kernel.h 299 backends/gpu/include/tfrt/gpu/passes/passes.h 56 backends/gpu/include/tfrt/gpu/system/system.h 68 backends/gpu/include/tfrt/gpu/tensor/dense_gpu_tensor.h 43 backends/gpu/include/tfrt/gpu/wrapper/blas_wrapper.h 62 backends/gpu/include/tfrt/gpu/wrapper/ccl_types.h 92 backends/gpu/include/tfrt/gpu/wrapper/ccl_wrapper.h 72 backends/gpu/include/tfrt/gpu/wrapper/cublas_wrapper.h 125 backends/gpu/include/tfrt/gpu/wrapper/cuda_forwards.h 51 backends/gpu/include/tfrt/gpu/wrapper/cuda_type_traits.h 43 backends/gpu/include/tfrt/gpu/wrapper/cuda_wrapper.h 234 backends/gpu/include/tfrt/gpu/wrapper/cudart_wrapper.h 79 backends/gpu/include/tfrt/gpu/wrapper/cudnn_wrapper.h 606 backends/gpu/include/tfrt/gpu/wrapper/cufft_wrapper.h 123 backends/gpu/include/tfrt/gpu/wrapper/cusolver_wrapper.h 74 backends/gpu/include/tfrt/gpu/wrapper/dense_map_utils.h 23 backends/gpu/include/tfrt/gpu/wrapper/dnn_wrapper.h 352 backends/gpu/include/tfrt/gpu/wrapper/driver_wrapper.h 315 backends/gpu/include/tfrt/gpu/wrapper/fft_wrapper.h 57 backends/gpu/include/tfrt/gpu/wrapper/hip_forwards.h 32 backends/gpu/include/tfrt/gpu/wrapper/hip_stub.h 166 backends/gpu/include/tfrt/gpu/wrapper/hip_wrapper.h 202 backends/gpu/include/tfrt/gpu/wrapper/hipfft_stub.h 18 backends/gpu/include/tfrt/gpu/wrapper/hipfft_wrapper.h 35 backends/gpu/include/tfrt/gpu/wrapper/miopen_stub.h 26 backends/gpu/include/tfrt/gpu/wrapper/miopen_wrapper.h 160 backends/gpu/include/tfrt/gpu/wrapper/nccl_wrapper.h 54 backends/gpu/include/tfrt/gpu/wrapper/rccl_stub.h 9 backends/gpu/include/tfrt/gpu/wrapper/rccl_wrapper.h 54 backends/gpu/include/tfrt/gpu/wrapper/rocblas_stub.h 21 backends/gpu/include/tfrt/gpu/wrapper/rocblas_wrapper.h 113 backends/gpu/include/tfrt/gpu/wrapper/rocsolver_stub.h 11 backends/gpu/include/tfrt/gpu/wrapper/rocsolver_wrapper.h 29 backends/gpu/include/tfrt/gpu/wrapper/runtime_wrapper.h 16 backends/gpu/include/tfrt/gpu/wrapper/solver_wrapper.h 25 backends/gpu/include/tfrt/gpu/wrapper/wrapper.h 323 backends/gpu/lib/core_runtime/gpu_op_handler.cc 112 backends/gpu/lib/core_runtime/gpu_op_registry.cc 34 backends/gpu/lib/core_runtime/gpu_op_registry_impl.h 14 backends/gpu/lib/core_runtime/op_handler_kernels.cc 26 backends/gpu/lib/core_runtime/op_handler_kernels.h 9 backends/gpu/lib/core_runtime/static_registration.cc 8 backends/gpu/lib/device/conversion_function.cc 126 backends/gpu/lib/device/device.cc 105 backends/gpu/lib/device/device_util.cc 26 backends/gpu/lib/device/eigen_support.cc 41 backends/gpu/lib/device/eigen_support.h 46 backends/gpu/lib/device/gpu_config.cc 53 backends/gpu/lib/device/static_registration.cc 9 backends/gpu/lib/gpu_entry_point.h 17 backends/gpu/lib/gpu_executor.cc 233 backends/gpu/lib/gpu_types.cc 174 backends/gpu/lib/kernels/blas_kernels.cc 202 backends/gpu/lib/kernels/ccl_kernels.cc 191 backends/gpu/lib/kernels/dnn_kernels.cc 612 backends/gpu/lib/kernels/driver_kernels.cc 337 backends/gpu/lib/kernels/gpu_ops.cc 306 backends/gpu/lib/kernels/solver_kernels.cc 143 backends/gpu/lib/kernels/static_registration.cc 17 backends/gpu/lib/memory/bfc_gpu_allocator.cc 215 backends/gpu/lib/ops/tf/binary_ops.cu.cc 33 backends/gpu/lib/ops/tf/dnn_ops.cc 628 backends/gpu/lib/ops/tf/dnn_ops.cu.cc 509 backends/gpu/lib/ops/tf/dnn_ops_cu.h 25 backends/gpu/lib/ops/tf/eigen_helper.cu.h 171 backends/gpu/lib/ops/tf/gpu_ops.cc 42 backends/gpu/lib/ops/tf/matmul_op.cc 134 backends/gpu/lib/ops/tf/matmul_op.h 15 backends/gpu/lib/ops/tf/mlir_ops.cc 226 backends/gpu/lib/ops/tf/nullary_ops.cc 59 backends/gpu/lib/ops/tf/pad_op.cc 148 backends/gpu/lib/ops/tf/pad_op.cu.cc 37 backends/gpu/lib/ops/tf/pad_op.h 37 backends/gpu/lib/ops/tf/pad_op_noncuda.cc 13 backends/gpu/lib/ops/tf/pad_op_noncuda.h 17 backends/gpu/lib/ops/tf/reduction_ops.cu.cc 382 backends/gpu/lib/ops/tf/static_registration.cc 4 backends/gpu/lib/ops/tf/transpose_op.cu.cc 236 backends/gpu/lib/ops/tf/unary_ops.cu.cc 128 backends/gpu/lib/passes/gpu_async_patterns.cc 141 backends/gpu/lib/passes/gpu_to_tfrt_passes.cc 1039 backends/gpu/lib/passes/set_entry_point.cc 45 backends/gpu/lib/system/system.cc 198 backends/gpu/lib/tensor/dense_gpu_tensor.cc 21 backends/gpu/lib/wrapper/blas_wrapper.cc 149 backends/gpu/lib/wrapper/ccl_enums.cc 110 backends/gpu/lib/wrapper/ccl_wrapper.cc 263 backends/gpu/lib/wrapper/cublas_enums.cc 394 backends/gpu/lib/wrapper/cublas_stub.cc 81 backends/gpu/lib/wrapper/cublas_wrapper.cc 149 backends/gpu/lib/wrapper/cuda_stub.cc 26 backends/gpu/lib/wrapper/cuda_wrapper.cc 656 backends/gpu/lib/wrapper/cudart_stub.cc 93 backends/gpu/lib/wrapper/cudart_wrapper.cc 89 backends/gpu/lib/wrapper/cudnn_enums.cc 193 backends/gpu/lib/wrapper/cudnn_stub.cc 32 backends/gpu/lib/wrapper/cudnn_wrapper.cc 1215 backends/gpu/lib/wrapper/cufft_stub.cc 21 backends/gpu/lib/wrapper/cufft_wrapper.cc 491 backends/gpu/lib/wrapper/cusolver_stub.cc 20 backends/gpu/lib/wrapper/cusolver_wrapper.cc 180 backends/gpu/lib/wrapper/dnn_wrapper.cc 838 backends/gpu/lib/wrapper/driver_wrapper.cc 1159 backends/gpu/lib/wrapper/fft_wrapper.cc 62 backends/gpu/lib/wrapper/hip_stub.cc 33 backends/gpu/lib/wrapper/hip_wrapper.cc 554 backends/gpu/lib/wrapper/hipfft_stub.cc 19 backends/gpu/lib/wrapper/hipfft_wrapper.cc 125 backends/gpu/lib/wrapper/miopen_enums.cc 142 backends/gpu/lib/wrapper/miopen_stub.cc 25 backends/gpu/lib/wrapper/miopen_wrapper.cc 326 backends/gpu/lib/wrapper/nccl_stub.cc 19 backends/gpu/lib/wrapper/nccl_wrapper.cc 134 backends/gpu/lib/wrapper/rccl_stub.cc 19 backends/gpu/lib/wrapper/rccl_wrapper.cc 135 backends/gpu/lib/wrapper/rocblas_enums.cc 207 backends/gpu/lib/wrapper/rocblas_stub.cc 19 backends/gpu/lib/wrapper/rocblas_wrapper.cc 127 backends/gpu/lib/wrapper/rocsolver_stub.cc 19 backends/gpu/lib/wrapper/rocsolver_wrapper.cc 56 backends/gpu/lib/wrapper/runtime_wrapper.cc 53 backends/gpu/lib/wrapper/solver_wrapper.cc 60 backends/gpu/lib/wrapper/symbol_loader.h 24 backends/gpu/lib/wrapper/wrapper_detail.h 107 backends/gpu/mlir_tests/conversion/async_conversion.mlir 48 backends/gpu/mlir_tests/conversion/async_streamify.mlir 46 backends/gpu/mlir_tests/conversion/async_to_tfrt.mlir 34 backends/gpu/mlir_tests/conversion/cast_streamify.mlir 43 backends/gpu/mlir_tests/conversion/entry_point.mlir 27 backends/gpu/mlir_tests/conversion/func_streamify.mlir 32 backends/gpu/mlir_tests/conversion/gpu_streamify.mlir 204 backends/gpu/mlir_tests/conversion/hoisting.mlir 65 backends/gpu/mlir_tests/conversion/ops.mlir 228 backends/gpu/mlir_tests/core_runtime/current_context_lifetime.mlir 69 backends/gpu/mlir_tests/core_runtime/get_device.mlir 43 backends/gpu/mlir_tests/core_runtime/logging_with_results.mlir 64 backends/gpu/mlir_tests/core_runtime/multi_gpu.mlir 94 backends/gpu/mlir_tests/core_runtime/op_handler.mlir 49 backends/gpu/mlir_tests/core_runtime/tensor_conversion.mlir 80 backends/gpu/mlir_tests/core_runtime/tf_addv2.mlir 170 backends/gpu/mlir_tests/core_runtime/tf_biasadd.benchmark.mlir 42 backends/gpu/mlir_tests/core_runtime/tf_biasadd.mlir 50 backends/gpu/mlir_tests/core_runtime/tf_cast.mlir 59 backends/gpu/mlir_tests/core_runtime/tf_const.mlir 32 backends/gpu/mlir_tests/core_runtime/tf_conv2d.benchmark.mlir 116 backends/gpu/mlir_tests/core_runtime/tf_conv2d.mlir 122 backends/gpu/mlir_tests/core_runtime/tf_fusedbatchnorm.benchmark.mlir 60 backends/gpu/mlir_tests/core_runtime/tf_fusedbatchnorm.mlir 85 backends/gpu/mlir_tests/core_runtime/tf_matmul.mlir 138 backends/gpu/mlir_tests/core_runtime/tf_maxpool.mlir 45 backends/gpu/mlir_tests/core_runtime/tf_mean.mlir 49 backends/gpu/mlir_tests/core_runtime/tf_pad.mlir 89 backends/gpu/mlir_tests/core_runtime/tf_reduction.benchmark.mlir 109 backends/gpu/mlir_tests/core_runtime/tf_reduction_ops.mlir 74 backends/gpu/mlir_tests/core_runtime/tf_relu.benchmark.mlir 39 backends/gpu/mlir_tests/core_runtime/tf_relu.mlir 52 backends/gpu/mlir_tests/core_runtime/tf_softmax.mlir 45 backends/gpu/mlir_tests/core_runtime/tf_tanh.mlir 33 backends/gpu/mlir_tests/core_runtime/tf_transpose.mlir 70 backends/gpu/mlir_tests/cuda/blas.mlir 158 backends/gpu/mlir_tests/cuda/ccl.mlir 45 backends/gpu/mlir_tests/cuda/dnn.mlir 163 backends/gpu/mlir_tests/cuda/error.mlir 37 backends/gpu/mlir_tests/cuda/event.mlir 35 backends/gpu/mlir_tests/cuda/executor.mlir 28 backends/gpu/mlir_tests/cuda/launch.mlir 117 backends/gpu/mlir_tests/cuda/memcpy.mlir 51 backends/gpu/mlir_tests/cuda/memset.mlir 37 backends/gpu/mlir_tests/cuda/module.mlir 75 backends/gpu/mlir_tests/cuda/register.mlir 41 backends/gpu/mlir_tests/cuda/solver.mlir 74 backends/gpu/mlir_tests/cuda/stream.mlir 70 backends/gpu/mlir_tests/cuda/while.mlir 117 backends/gpu/tools/stub_codegen/header_codegen.py 64 backends/gpu/tools/stub_codegen/impl_codegen.py 33 backends/gpu/tools/tfrt_gpu_executor/main.cc 71 backends/gpu/tools/tfrt_gpu_opt/tfrt_gpu_opt.cc 104 backends/gpu/tools/tfrt_gpu_translate/static_registration.cc 19