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

  /external/tensorflow/tensorflow/core/kernels/
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...]
fused_batch_norm_op.cc 511 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_),
569 tensor_format_, is_training_);
574 TensorFormat tensor_format_; member in class:tensorflow::FusedBatchNormOp
588 OP_REQUIRES(context, FormatFromString(tensor_format, &tensor_format_),
660 offset_backprop, tensor_format_);
664 CHECK(tensor_format_ == FORMAT_NHWC)
685 TensorFormat tensor_format_; member in class:tensorflow::FusedBatchNormGradOp
    [all...]

Completed in 586 milliseconds