|
..
|
|
kernels
|
|
BUILD.bazel
|
|
argminmax_impl.h
|
|
backend.cc
|
|
backend.h
|
|
backend_test.cc
|
|
batch_mat_mul_impl.cc
|
|
batch_mat_mul_impl.h
|
|
binary.cc
|
|
binary.h
|
|
bincount_impl.h
|
|
build_defs.bzl
|
|
check_macros.h
|
|
clamp_impl.cc
|
|
clamp_impl.h
|
|
conv2d_impl.cc
|
|
conv2d_impl.h
|
|
conv3d_impl.h
|
|
dilation2d_impl.h
|
|
elu_impl.cc
|
|
elu_impl.h
|
|
interpolate_bilinear_impl.cc
|
|
interpolate_bilinear_impl.h
|
|
leakyrelu_impl.cc
|
|
leakyrelu_impl.h
|
|
non_max_suppression_impl.cc
|
|
non_max_suppression_impl.h
|
|
pool3d_impl.h
|
|
post.js
|
|
pre.js
|
|
prelu_impl.cc
|
|
prelu_impl.h
|
|
scatter_impl.cc
|
|
scatter_impl.h
|
|
shape.h
|
|
sigmoid_impl.cc
|
|
sigmoid_impl.h
|
|
sin_cos_workaround.cc
|
|
sin_cos_workaround.h
|
|
transpose_impl.cc
|
|
transpose_impl.h
|
|
unary.cc
|
|
unary.h
|
|
util.cc
|
|
util.h
|
|
util_test.cc
|