Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for IsInPlaceOp (0.71 sec)

  1. tensorflow/compiler/jit/clone_constants_for_better_clustering.cc

    // pressure on GPUs.
    absl::StatusOr<bool> IsSmallConstant(Node* n) {
      if (!n->IsConstant()) {
        return false;
      }
    
      return IsConstantSmall(n);
    }
    
    bool IsInPlaceOp(absl::string_view op_name) {
      return op_name == "InplaceUpdate" || op_name == "InplaceAdd" ||
             op_name == "InplaceSub";
    }
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.3K bytes
    - Viewed (0)
Back to top