HomeSort by relevance Sort by last modified time
    Searched refs:FusedBatchNormOp (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
batch_norm_op.cc 29 class FusedBatchNormOp : public XlaOpKernel {
31 explicit FusedBatchNormOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {
109 REGISTER_XLA_OP(Name("FusedBatchNorm"), FusedBatchNormOp);
110 REGISTER_XLA_OP(Name("FusedBatchNormV2"), FusedBatchNormOp);
  /external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.cc 39 // Functor used by FusedBatchNormOp to do the computations.
524 class FusedBatchNormOp : public OpKernel {
526 explicit FusedBatchNormOp(OpKernelConstruction* context) : OpKernel(context) {
712 FusedBatchNormOp<CPUDevice, float, float>);
722 FusedBatchNormOp<CPUDevice, float, float>);
    [all...]

Completed in 120 milliseconds