Home
last modified time | relevance | path

Searched defs:FusedBatchNormOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc31 class FusedBatchNormOp : public XlaOpKernel { class
33 explicit FusedBatchNormOp(OpKernelConstruction* ctx) in FusedBatchNormOp() function in tensorflow::__anon73cf47b80111::FusedBatchNormOp
36 FusedBatchNormOp(OpKernelConstruction* ctx, bool is_batch_norm_ex) in FusedBatchNormOp() function in tensorflow::__anon73cf47b80111::FusedBatchNormOp
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc1346 class FusedBatchNormOp : public FusedBatchNormOpBase<Device, T, U> { class
1348 explicit FusedBatchNormOp(OpKernelConstruction* context) in FusedBatchNormOp() function in tensorflow::FusedBatchNormOp