- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for LookupTableImportV2Op (0.97 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
return success(); } }; class LegalizeHashTableImportOpPattern : public OpRewritePattern<TF::LookupTableImportV2Op> { public: using OpRewritePattern<TF::LookupTableImportV2Op>::OpRewritePattern; LogicalResult matchAndRewrite(TF::LookupTableImportV2Op import_op, PatternRewriter& rewriter) const override { auto handle_op = import_op.getTableHandle().getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/init_text_file_to_import.cc
lines)); Value value_constant_tensor = rewriter.create<arith::ConstantOp>( op.getLoc(), rewriter.getI64TensorAttr(line_nums)); // Replace the given op with LookupTableImportV2Op. rewriter.create<LookupTableImportV2Op>(op.getLoc(), op.getTableHandle(), key_constant_tensor, value_constant_tensor); rewriter.eraseOp(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/analyze_variables.cc
// supported by TFLite. bool IsSupportedTFLiteResourceOp(Operation* op) { return llvm::isa<TF::ReadVariableOp, TF::AssignVariableOp, TF::VarHandleOp, TF::LookupTableFindV2Op, TF::LookupTableImportV2Op, TF::LookupTableSizeV2Op>(op); } // Returns true if 'op' is TF/TFLite control flow op that can accept resource // type. Usually these ops are just pass through, they call another subgraph and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
std::unique_ptr<OperationPass<func::FuncOp>> CreateDeviceIndexSelectorPass(); // Creates function pass to replace InitializeTableFromTextFileV2Ops with // LookupTableImportV2Op ops. std::unique_ptr<OperationPass<func::FuncOp>> CreateInitTextFileToImportPass( std::string saved_model_dir = ""); // Creates function pass to cluster TensorFlow ops by host. The program
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
not required. We can use the output of first replica in such cases. ### `-tf-init-text-file-to-import` _Convert InitializeTableFromTextFileV2 ops to LookupTableImportV2Op to remove the dependency on asset files_ #### Options ```
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0)