- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for PartitionedCallOp (0.11 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
} if (skip_resize_) { target.addLegalOp<TF::ResizeBilinearOp>(); target.addLegalOp<TF::ResizeNearestNeighborOp>(); } if (skip_partitioned_calls_) { target.addLegalOp<TF::PartitionedCallOp>(); target.addLegalOp<TF::StatefulPartitionedCallOp>(); } FrozenRewritePatternSet frozen_patterns(std::move(patterns)); if (failed(applyPartialConversion(func, target, frozen_patterns))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
auto function_uses = SymbolTable::getSymbolUses(func, &module.getBodyRegion()); if (!function_uses.has_value()) return false; for (auto& function_use : function_uses.value()) { if (!llvm::isa<TF::PartitionedCallOp, TF::StatefulPartitionedCallOp>( function_use.getUser())) { return false; } } return true; } // Returns the `shared_name` attribute value if exists. If not, returns an
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0)