mkl_fused_batch_norm_op.cc | 56 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), 106 mkl_context.MklExtractParams(context, tensor_format_); 138 tensor_format_); 211 TensorFormat tensor_format_; member in class:tensorflow::MklFusedBatchNormOp 353 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), 394 mkl_context.MklExtractParams(context, tensor_format_); 425 tensor_format_); 490 TensorFormat tensor_format_; member in class:tensorflow::MklFusedBatchNormGradOp 698 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_), 933 TensorFormat tensor_format_; member in class:tensorflow::MklFusedBatchNormOp 1282 TensorFormat tensor_format_; member in class:tensorflow::MklFusedBatchNormGradOp [all...] |