HomeSort by relevance Sort by last modified time
    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