Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for WriteLocalVariable (0.15 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.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