Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for aggregate (0.19 sec)

  1. tensorflow/c/eager/tape.h

      } else {
        std::array<Gradient*, 2> to_aggregate;
        to_aggregate[0] = tangent;
        to_aggregate[1] = existing->second;
        // AggregateGradients steals a reference to each of its arguments. We
        // MarkAsResult on `tangent` above so we don't steal a reference to it.
        existing->second = vspace_.AggregateGradients(to_aggregate);
      }
    }
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  2. tensorflow/c/eager/gradients.cc

      // and not for correctness. The only downside of keeping this 1 seems to be
      // that the gradient accumulation is unbounded and we will never
      // aggressively aggregate accumulated gradients to recover memory.
      // Revisit and fix.
      return 1;
    }
    
    // Consumes references to the tensors in the gradient_tensors list and returns
    // a tensor with the result.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  3. RELEASE.md

            `MultiWorkerMirroredStrategy`.
    *   `tf.keras`:
        *   Added `experimental_aggregate_gradients` argument to
            `tf.keras.optimizer.Optimizer.apply_gradients`. This allows custom
            gradient aggregation and processing aggregated gradients in custom
            training loop.
        *   Allow `pathlib.Path` paths for loading models via Keras API.
    *   `tf.function`/AutoGraph:
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Mon Apr 29 19:17:57 GMT 2024
    - 727.7K bytes
    - Viewed (8)
  4. tensorflow/BUILD

            "//tensorflow/compiler/mlir/quantization/stablehlo/python:pywrap_quantization_lib_impl",
            "//tensorflow/compiler/mlir/quantization/tensorflow/calibrator:custom_aggregator_op",
            "//tensorflow/compiler/mlir/quantization/tensorflow/python:quantize_model_cc_impl",
            "//tensorflow/compiler/mlir/quantization/tensorflow:passes",
            "//tensorflow/compiler/mlir/tensorflow",
    Plain Text
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 09 18:15:11 GMT 2024
    - 53.4K bytes
    - Viewed (8)
Back to top