Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for IsSupportedTFLiteResourceOp (0.36 sec)

  1. tensorflow/compiler/mlir/lite/transforms/analyze_variables.cc

    // variables should be legalized to TFLite or not.
    const char kLegalizeTflVariables[] = "tfl._legalize_tfl_variables";
    
    // Returns true if 'op' is TF op that accepts resource type, but is
    // supported by TFLite.
    bool IsSupportedTFLiteResourceOp(Operation* op) {
      return llvm::isa<TF::ReadVariableOp, TF::AssignVariableOp, TF::VarHandleOp,
                       TF::LookupTableFindV2Op, TF::LookupTableImportV2Op,
                       TF::LookupTableSizeV2Op>(op);
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.3K bytes
    - Viewed (0)
Back to top