- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for IteratorGetNext (0.48 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
For example, original program: ```mlir module { func @while_body { %input = "tf.IteratorGetNext"(...) {device = "/CPU:0"}: -> tensor<2x224x224x3xf32> %device_launch = "tf_device.cluster_func"(%input,...) {func = @_func,...) return ... }
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/tensorflow/tests/side-effect-analysis-test.mlir
// expected-remark@above {{ID: 1}} // expected-remark@above {{Successors: {2}}} %0 = "tf.IteratorGetNext"(%handle) {_class = ["loc:@GeneratorDataset_2"], device = "/job:tpu_host_worker/replica:0/task:0/device:CPU:0"} : (tensor<!tf_type.resource>) -> tensor<!tf_type.string> // expected-remark@above {{ID: 2}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
const auto& element_type = element_types->list().type(idx); return ConvertToMlirTensorType(output_shape, element_type, &builder); }; if (node.type_string() == "IteratorGetNext" || node.type_string() == "IteratorGetNextSync" || node.type_string() == "MultiDeviceIteratorGetNextFromShard") return type_from_array_attr("output_shapes", "output_types");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)