Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for getZp (0.38 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc

    // failure.
    template <typename OpT>
    LogicalResult GetOp(Region* region, OpT* result) {
      *result = {};
      for (auto op : region->getOps<OpT>()) {
        if (*result) return op.emitError("should be unique within a function");
        *result = op;
      }
      return success();
    }
    
    LogicalResult RunOnRegion(Region* region) {
      RecvTPUEmbeddingActivationsOp recv_op;
      if (failed(GetOp(region, &recv_op))) return failure();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 22:55:42 UTC 2024
    - 4.5K bytes
    - Viewed (0)
Back to top