- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for IsSmallConstant (0.19 sec)
-
tensorflow/compiler/jit/clone_constants_for_better_clustering.cc
const int kSmallTensorThreshold = 16; return total_elements < kSmallTensorThreshold; } // We only clone small constants since we want to avoid increasing memory // pressure on GPUs. absl::StatusOr<bool> IsSmallConstant(Node* n) { if (!n->IsConstant()) { return false; } return IsConstantSmall(n); } bool IsInPlaceOp(absl::string_view op_name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 7.3K bytes - Viewed (0)