Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for _TPUDeviceOrdinalPlaceholder (0.23 sec)

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

        return region && region->isProperAncestor(replicate_region);
      };
    
      for (Value operand : op->getOperands())
        if (!ancestor_of_replicate(operand.getParentRegion())) return false;
    
      // _TPUDeviceOrdinalPlaceholder implicitly depends on the replica.
      if (llvm::isa<TF::_TPUDeviceOrdinalPlaceholderOp>(op)) return false;
    
      bool has_replicate_operands = false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

        TF_StrTensor:$summary_metadata
      );
    
      let results = (outs);
    
      TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<2>;
    }
    
    def TF__TPUDeviceOrdinalPlaceholderOp : TF_Op<"_TPUDeviceOrdinalPlaceholder", [Pure]> {
      let summary = [{
    Placeholder for a device ordinal that depends on its tf_device.replicate ancestor.
      }];
    
      let description = [{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
Back to top