Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for CloneVarHandleOpsIntoSaveFunc (0.28 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc

      return Create1DStringConst(shape_and_slices_values, loc, builder);
    }
    
    // Returns cloned `VarHandleOp`s. Assumes `save_func`'s body is empty.
    SmallVector<TF::VarHandleOp> CloneVarHandleOpsIntoSaveFunc(
        func::FuncOp save_func, const ArrayRef<TF::VarHandleOp> var_handle_ops) {
      Block& save_op_block = save_func.getBody().front();
    
      IRMapping mapper{};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 9.5K bytes
    - Viewed (0)
Back to top