OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:BiasOp
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
bias_ops.cc
28
class
BiasOp
: public XlaOpKernel {
30
explicit
BiasOp
(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {
72
REGISTER_XLA_OP(Name("BiasAdd"),
BiasOp
);
73
REGISTER_XLA_OP(Name("BiasAddV1"),
BiasOp
);
/external/tensorflow/tensorflow/core/kernels/
bias_op.cc
85
class
BiasOp
: public BinaryOp<T> {
87
explicit
BiasOp
(OpKernelConstruction* context) : BinaryOp<T>(context) {
181
BiasOp
<CPUDevice, type>); \
184
BiasOp
<CPUDevice, type>);
193
BiasOp
<SYCLDevice, type>); \
196
BiasOp
<SYCLDevice, type>);
308
class
BiasOp
<GPUDevice, T> : public BinaryOp<T> {
311
explicit
BiasOp
(OpKernelConstruction* context) : BinaryOp<T>(context) {
358
BiasOp
<GPUDevice, type>); \
361
BiasOp
<GPUDevice, type>)
[
all
...]
Completed in 286 milliseconds