Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for BroadcastGradientArgs (0.44 sec)

  1. tensorflow/cc/gradients/math_grad.cc

      Output s_max = Shape(scope, max);
    
      Output min_mask = Less(scope, x, min);
      Output max_mask = Greater(scope, x, max);
    
      auto r_min = internal::BroadcastGradientArgs(scope, s_x, s_min);
      auto r_max = internal::BroadcastGradientArgs(scope, s_x, s_max);
    
      Output grad = grad_inputs[0];
      Output zeros = ZerosLike(scope, grad);
    
      Output x_grad =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 25 18:20:20 UTC 2023
    - 50.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass.cc

          "Range", "Rank", "Reshape", "Shape", "ShapeN", "Size", "Squeeze",
          "Transpose", "ZerosLike", "OnesLike", "BiasAdd" /*PW + Broadcast*/,
          "BroadcastArgs", "BroadcastGradientArgs", "OneHot", "Concat", "ConcatV2",
          "ConcatOffset", "Const", "MirrorPad", "MirrorPadGrad", "Pack", "Pad",
          "PadV2", "Reverse", "ReverseV2", "ReverseSequence", "Slice", "Split",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
Back to top