- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for aggregate (0.19 sec)
-
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) -
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) -
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) -
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)