ntt123's picture
add sparse matmul
21f3d42
raw
history blame
612 Bytes
# [internal] load placeholder
licenses(["notice"])
cc_library(
name = "sparse_matmul",
hdrs = [
"sparse_matmul.h",
],
visibility = ["//visibility:public"],
deps = [
"//sparse_matmul/compute:gru_gates",
"//sparse_matmul/layers:layer",
"//sparse_matmul/layers:matrix",
"//sparse_matmul/layers:utils",
"//sparse_matmul/numerics:fast_transcendentals",
"//sparse_matmul/numerics:types",
"//sparse_matmul/os:coop_threads",
"//sparse_matmul/vector:cache_aligned_vector",
], # internal :sparse_matmul deps placeholder
)