Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for WriteLocalVariable (0.2 sec)

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

          stack.getLoc(), ArrayRef<Type>{size_var_type}, ArrayRef<Value>{});
      // Zero-initialize the local vars.
      cutil::WriteLocalVariable(local_size_var,
                                cutil::GetR1Const({0LL}, builder, stack.getLoc()),
                                builder, stack.getLoc());
      cutil::WriteLocalVariable(local_var, buffer, builder, stack.getLoc());
      stack.getHandle().replaceAllUsesWith(local_var);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.h

    // Creates a ReadVariableOp on a local variable.
    Value ReadLocalVariable(Value local_var, OpBuilder builder, Location loc);
    
    // Creates an AssignVariableOp on a local variable.
    TF::AssignVariableOp WriteLocalVariable(Value local_var, Value value,
                                            OpBuilder builder, Location loc);
    
    // Adds two values, or creates a logical-or if they are boolean type.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 4.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc

                  ta.getContext()));
      auto local_var = builder.create<TF::MlirLocalVarOp>(
          ta.getLoc(), ArrayRef<Type>{var_type}, ArrayRef<Value>{});
      cutil::WriteLocalVariable(local_var, buffer, builder, ta.getLoc());
      ta.getHandle().replaceAllUsesWith(local_var);
      // The flow output is just a way for the front end to enforce ordering among
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 02 20:41:19 UTC 2023
    - 40.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/collection_ops_util.cc

                                 .getSubtypes()[0]},
              ArrayRef<Value>{local_var})
          .getValue();
    }
    
    // Creates an AssignVariableOp on a local variable.
    TF::AssignVariableOp WriteLocalVariable(Value local_var, Value value,
                                            OpBuilder builder, Location loc) {
      return builder.create<TF::AssignVariableOp>(
          loc, ArrayRef<Type>{}, ArrayRef<Value>{local_var, value});
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.5K bytes
    - Viewed (0)
Back to top