Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TPUResourceReadForWritePass (0.35 sec)

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

    // associated resource read, so the resource can later be fused into TPUExecute.
    namespace {
    struct TPUResourceReadForWritePass
        : public impl::TPUResourceReadForWritePassBase<
              TPUResourceReadForWritePass> {
      void runOnOperation() override;
    };
    
    // Helper struct holding a resource value and its associated type.
    struct ResourceValueAndSubtype {
      Value resource;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 16:54:40 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

        to XLA compiler for input/output buffer space optimizations.
      }];
    
      let constructor = "TFDevice::CreateMarkInputOutputAliasesPass()";
    }
    
    def TPUResourceReadForWritePass : Pass<"tf-tpu-resource-read-for-write", "ModuleOp"> {
      let summary = "Inserts tf.ReadVariableOp inputs to a TPU cluster for resource writes with no reads";
    
      let description = [{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
Back to top