Path	Lines of Code
tfjs-backend-webgpu/benchmarks/config.js	17
tfjs-backend-webgpu/benchmarks/index.html	51
tfjs-backend-webgpu/benchmarks/main.css	183
tfjs-backend-webgpu/benchmarks/main.js	132
tfjs-backend-webgpu/benchmarks/util.js	122
tfjs-backend-webgpu/rollup.config.js	93
tfjs-backend-webgpu/src/FromPixelsExternalImage.ts	54
tfjs-backend-webgpu/src/activation_util.ts	23
tfjs-backend-webgpu/src/addn_packed_webgpu.ts	47
tfjs-backend-webgpu/src/argminmax_webgpu.ts	121
tfjs-backend-webgpu/src/backend_webgpu.ts	716
tfjs-backend-webgpu/src/backend_webgpu_test_registry.ts	9
tfjs-backend-webgpu/src/batchnorm_webgpu.ts	64
tfjs-backend-webgpu/src/binary_op_complex_webgpu.ts	42
tfjs-backend-webgpu/src/binary_op_shared_webgpu.ts	79
tfjs-backend-webgpu/src/binary_op_util.ts	186
tfjs-backend-webgpu/src/binary_op_vec4_webgpu.ts	45
tfjs-backend-webgpu/src/binary_op_webgpu.ts	41
tfjs-backend-webgpu/src/binary_ops.ts	22
tfjs-backend-webgpu/src/buffer_manager.ts	94
tfjs-backend-webgpu/src/clip_vec4_webgpu.ts	42
tfjs-backend-webgpu/src/clip_webgpu.ts	37
tfjs-backend-webgpu/src/concat_webgpu.ts	63
tfjs-backend-webgpu/src/constants.ts	5
tfjs-backend-webgpu/src/conv2d_mm_vec4_webgpu.ts	213
tfjs-backend-webgpu/src/conv2d_mm_webgpu.ts	151
tfjs-backend-webgpu/src/conv2d_naive_webgpu.ts	109
tfjs-backend-webgpu/src/conv_backprop_mm_webgpu.ts	88
tfjs-backend-webgpu/src/conv_backprop_webgpu.ts	75
tfjs-backend-webgpu/src/crop_and_resize_webgpu.ts	114
tfjs-backend-webgpu/src/depth_to_space_webgpu.ts	79
tfjs-backend-webgpu/src/depthwise_conv2d_3x3_webgpu.ts	126
tfjs-backend-webgpu/src/depthwise_conv2d_webgpu.ts	128
tfjs-backend-webgpu/src/fill_webgpu.ts	30
tfjs-backend-webgpu/src/flags_webgpu.ts	12
tfjs-backend-webgpu/src/flip_left_right_webgpu.ts	32
tfjs-backend-webgpu/src/gather_nd_webgpu.ts	46
tfjs-backend-webgpu/src/gather_webgpu.ts	46
tfjs-backend-webgpu/src/im2col_webgpu.ts	57
tfjs-backend-webgpu/src/index.ts	33
tfjs-backend-webgpu/src/kernel_utils/int.ts	9
tfjs-backend-webgpu/src/kernel_utils/kernel_funcs_utils.ts	122
tfjs-backend-webgpu/src/kernel_utils/reduce.ts	52
tfjs-backend-webgpu/src/kernel_utils/shared.ts	73
tfjs-backend-webgpu/src/kernels/Abs.ts	11
tfjs-backend-webgpu/src/kernels/Add.ts	14
tfjs-backend-webgpu/src/kernels/AddN.ts	22
tfjs-backend-webgpu/src/kernels/ArgMax.ts	34
tfjs-backend-webgpu/src/kernels/ArgMin.ts	34
tfjs-backend-webgpu/src/kernels/AvgPool.ts	43
tfjs-backend-webgpu/src/kernels/BatchMatMul.ts	18
tfjs-backend-webgpu/src/kernels/BatchMatMul_impl.ts	106
tfjs-backend-webgpu/src/kernels/BatchToSpaceND.ts	52
tfjs-backend-webgpu/src/kernels/Cast.ts	54
tfjs-backend-webgpu/src/kernels/Ceil.ts	11
tfjs-backend-webgpu/src/kernels/ClipByValue.ts	29
tfjs-backend-webgpu/src/kernels/Complex.ts	19
tfjs-backend-webgpu/src/kernels/Concat.ts	28
tfjs-backend-webgpu/src/kernels/Concat_impl.ts	84
tfjs-backend-webgpu/src/kernels/Conv2D.ts	20
tfjs-backend-webgpu/src/kernels/Conv2DBackpropInput.ts	55
tfjs-backend-webgpu/src/kernels/Conv2D_impl.ts	243
tfjs-backend-webgpu/src/kernels/Cos.ts	9
tfjs-backend-webgpu/src/kernels/Cosh.ts	9
tfjs-backend-webgpu/src/kernels/CropAndResize.ts	22
tfjs-backend-webgpu/src/kernels/DepthToSpace.ts	18
tfjs-backend-webgpu/src/kernels/DepthwiseConv2dNative.ts	39
tfjs-backend-webgpu/src/kernels/Einsum.ts	75
tfjs-backend-webgpu/src/kernels/Elu.ts	9
tfjs-backend-webgpu/src/kernels/Equal.ts	11
tfjs-backend-webgpu/src/kernels/Exp.ts	14
tfjs-backend-webgpu/src/kernels/ExpandDims.ts	29
tfjs-backend-webgpu/src/kernels/Expm1.ts	11
tfjs-backend-webgpu/src/kernels/Fill.ts	24
tfjs-backend-webgpu/src/kernels/FlipLeftRight.ts	16
tfjs-backend-webgpu/src/kernels/Floor.ts	11
tfjs-backend-webgpu/src/kernels/FloorDiv.ts	10
tfjs-backend-webgpu/src/kernels/FromPixels.ts	88
tfjs-backend-webgpu/src/kernels/FromPixels_utils/from_pixels_import_webgpu.ts	38
tfjs-backend-webgpu/src/kernels/FromPixels_utils/from_pixels_webgpu.ts	136
tfjs-backend-webgpu/src/kernels/FusedBatchNorm.ts	28
tfjs-backend-webgpu/src/kernels/FusedConv2D.ts	32
tfjs-backend-webgpu/src/kernels/FusedDepthwiseConv2D.ts	54
tfjs-backend-webgpu/src/kernels/GatherNd.ts	20
tfjs-backend-webgpu/src/kernels/GatherV2.ts	67
tfjs-backend-webgpu/src/kernels/Greater.ts	14
tfjs-backend-webgpu/src/kernels/GreaterEqual.ts	14
tfjs-backend-webgpu/src/kernels/Identity.ts	14
tfjs-backend-webgpu/src/kernels/Imag.ts	15
tfjs-backend-webgpu/src/kernels/LeakyRelu.ts	21
tfjs-backend-webgpu/src/kernels/Less.ts	11
tfjs-backend-webgpu/src/kernels/LessEqual.ts	14
tfjs-backend-webgpu/src/kernels/Log.ts	11
tfjs-backend-webgpu/src/kernels/LogicalAnd.ts	12
tfjs-backend-webgpu/src/kernels/LogicalNot.ts	9
tfjs-backend-webgpu/src/kernels/Max.ts	16
tfjs-backend-webgpu/src/kernels/MaxPool.ts	44
tfjs-backend-webgpu/src/kernels/Maximum.ts	13
tfjs-backend-webgpu/src/kernels/Mean.ts	16
tfjs-backend-webgpu/src/kernels/Min.ts	16
tfjs-backend-webgpu/src/kernels/Minimum.ts	13
tfjs-backend-webgpu/src/kernels/MirrorPad.ts	19
tfjs-backend-webgpu/src/kernels/Multiply.ts	14
tfjs-backend-webgpu/src/kernels/Neg.ts	23
tfjs-backend-webgpu/src/kernels/NonMaxSuppressionV3.ts	25
tfjs-backend-webgpu/src/kernels/NonMaxSuppressionV5.ts	36
tfjs-backend-webgpu/src/kernels/NotEqual.ts	14
tfjs-backend-webgpu/src/kernels/OnesLike.ts	33
tfjs-backend-webgpu/src/kernels/Pack.ts	39
tfjs-backend-webgpu/src/kernels/PadV2.ts	34
tfjs-backend-webgpu/src/kernels/Pow.ts	11
tfjs-backend-webgpu/src/kernels/Prelu.ts	16
tfjs-backend-webgpu/src/kernels/Prod.ts	16
tfjs-backend-webgpu/src/kernels/Range.ts	15
tfjs-backend-webgpu/src/kernels/Real.ts	15
tfjs-backend-webgpu/src/kernels/RealDiv.ts	9
tfjs-backend-webgpu/src/kernels/Relu.ts	9
tfjs-backend-webgpu/src/kernels/Relu6.ts	9
tfjs-backend-webgpu/src/kernels/Reshape.ts	24
tfjs-backend-webgpu/src/kernels/ResizeBilinear.ts	28
tfjs-backend-webgpu/src/kernels/ResizeNearestNeighbor.ts	29
tfjs-backend-webgpu/src/kernels/RotateWithOffset.ts	31
tfjs-backend-webgpu/src/kernels/Rsqrt.ts	11
tfjs-backend-webgpu/src/kernels/ScatterNd.ts	16
tfjs-backend-webgpu/src/kernels/Select.ts	17
tfjs-backend-webgpu/src/kernels/Sigmoid.ts	9
tfjs-backend-webgpu/src/kernels/Sin.ts	9
tfjs-backend-webgpu/src/kernels/Sinh.ts	9
tfjs-backend-webgpu/src/kernels/Slice.ts	30
tfjs-backend-webgpu/src/kernels/Softmax.ts	45
tfjs-backend-webgpu/src/kernels/SpaceToBatchND.ts	55
tfjs-backend-webgpu/src/kernels/SparseToDense.ts	36
tfjs-backend-webgpu/src/kernels/SplitV.ts	28
tfjs-backend-webgpu/src/kernels/Sqrt.ts	9
tfjs-backend-webgpu/src/kernels/Square.ts	14
tfjs-backend-webgpu/src/kernels/SquaredDifference.ts	11
tfjs-backend-webgpu/src/kernels/StridedSlice.ts	74
tfjs-backend-webgpu/src/kernels/StringNGrams.ts	33
tfjs-backend-webgpu/src/kernels/Sub.ts	14
tfjs-backend-webgpu/src/kernels/Sum.ts	16
tfjs-backend-webgpu/src/kernels/Tanh.ts	9
tfjs-backend-webgpu/src/kernels/Tile.ts	29
tfjs-backend-webgpu/src/kernels/TopK.ts	130
tfjs-backend-webgpu/src/kernels/Transform.ts	49
tfjs-backend-webgpu/src/kernels/Transpose.ts	37
tfjs-backend-webgpu/src/kernels/Unpack.ts	45
tfjs-backend-webgpu/src/kernels/ZerosLike.ts	37
tfjs-backend-webgpu/src/kernels/_FusedMatMul.ts	28
tfjs-backend-webgpu/src/matmul_packed_vec4_webgpu.ts	198
tfjs-backend-webgpu/src/matmul_packed_webgpu.ts	295
tfjs-backend-webgpu/src/matmul_reduce.ts	135
tfjs-backend-webgpu/src/matmul_small_output_size_webgpu.ts	114
tfjs-backend-webgpu/src/mirror_pad_webgpu.ts	66
tfjs-backend-webgpu/src/pad_webgpu.ts	59
tfjs-backend-webgpu/src/pool2d_webgpu.ts	64
tfjs-backend-webgpu/src/pool_filtersizeone_webgpu.ts	38
tfjs-backend-webgpu/src/reduce_webgpu.ts	97
tfjs-backend-webgpu/src/register_all_kernels.ts	204
tfjs-backend-webgpu/src/resize_bilinear_webgpu.ts	60
tfjs-backend-webgpu/src/resize_nearest_neighbor_webgpu.ts	62
tfjs-backend-webgpu/src/rotate_webgpu.ts	59
tfjs-backend-webgpu/src/scatter_optimized_webgpu.ts	98
tfjs-backend-webgpu/src/scatter_webgpu.ts	85
tfjs-backend-webgpu/src/select_webgpu.ts	60
tfjs-backend-webgpu/src/shader_preprocessor.ts	583
tfjs-backend-webgpu/src/shader_util.ts	14
tfjs-backend-webgpu/src/slice_webgpu.ts	63
tfjs-backend-webgpu/src/strided_slice_webgpu.ts	52
tfjs-backend-webgpu/src/tile_webgpu.ts	51
tfjs-backend-webgpu/src/top_k_webgpu.ts	167
tfjs-backend-webgpu/src/transform_webgpu.ts	138
tfjs-backend-webgpu/src/transpose_shared_webgpu.ts	49
tfjs-backend-webgpu/src/transpose_webgpu.ts	56
tfjs-backend-webgpu/src/unary_op_util.ts	149
tfjs-backend-webgpu/src/unary_op_webgpu.ts	38
tfjs-backend-webgpu/src/webgpu.ts	4
tfjs-backend-webgpu/src/webgpu_program.ts	59
tfjs-backend-webgpu/src/webgpu_util.ts	102