- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for AnonymousIteratorV3Op (0.19 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
registry.insert<tf_device::TensorFlowDeviceDialect>(); } void runOnOperation() override; }; // Create the AnonymousIterator for `reduce_dataset` with `dataset_types` using // `builder`. AnonymousIteratorV3Op CreateIterator(OpBuilder builder, llvm::ArrayRef<Type> dataset_types, ReduceDatasetOp reduce_dataset) { llvm::SmallVector<Attribute, 2> shape_attrs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/name_anonymous_iterators.cc
[&](TF::AnonymousIteratorOp op) { count = replace(op, count); }); getOperation().walk( [&](TF::AnonymousIteratorV2Op op) { count = replace(op, count); }); getOperation().walk( [&](TF::AnonymousIteratorV3Op op) { count = replace(op, count); }); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateNameAnonymousIteratorsPass() { return std::make_unique<NameAnonymousIteratorsPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 2.7K bytes - Viewed (0)