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

  /external/tensorflow/tensorflow/core/profiler/g3doc/
profile_memory.md 27 FusedBatchNormGrad 10802.14MB (72.98%, 8.98%)
  /external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.cc 45 struct FusedBatchNormGrad;
154 struct FusedBatchNormGrad<CPUDevice, T, U> {
162 errors::Internal("The CPU implementation of FusedBatchNormGrad "
384 struct FusedBatchNormGrad<GPUDevice, T, U> {
398 VLOG(2) << "FusedBatchNormGrad:"
678 functor::FusedBatchNormGrad<Device, T, U>()(
686 << "The implementation of FusedBatchNormGrad with is_training=False "
715 Name("FusedBatchNormGrad").Device(DEVICE_CPU).TypeConstraint<float>("T"),
    [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
layout_optimizer_test.cc 212 ops::FusedBatchNormGrad::Attrs attrs;
215 ops::FusedBatchNormGrad(s->WithOpName("FusedBatchNormGrad"), y_backprop,
532 auto conv_node = node_map.GetNode("FusedBatchNormGrad");
546 auto conv_node = node_map.GetNode("FusedBatchNormGrad");
    [all...]

Completed in 68 milliseconds