Searched refs:batch_dataset (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | hoist_random_uniform.cc | 91 NodeDef batch_dataset; in MakeBatchTwo() local
92 batch_dataset.set_op("BatchDatasetV2"); in MakeBatchTwo()
94 &batch_dataset); in MakeBatchTwo()
97 batch_dataset.add_input(random_dataset.name()); in MakeBatchTwo()
98 batch_dataset.add_input(batch_size->name()); in MakeBatchTwo()
99 batch_dataset.add_input(drop_reminder->name()); in MakeBatchTwo()
101 (*batch_dataset.mutable_attr())["output_shapes"] in MakeBatchTwo()
107 (*batch_dataset.mutable_attr())["output_types"].mutable_list()->add_type( in MakeBatchTwo()
110 return batch_dataset; in MakeBatchTwo()
267 const auto* batch_dataset = in OptimizeAndCollectStats() local
[all …]
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/ |
| D | mnist_conv_custom_training_benchmark_test.py | 89 def distributed_train_step(self, batch_dataset, model, loss_fn, optimizer, argument
108 batch_dataset,
154 for batch_dataset in train_dataset:
159 total_loss += self.distributed_train_step(batch_dataset, model,
164 total_loss += self.train_step(batch_dataset, model, loss_fn,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
| D | tf-data-pipeline.pbtxt | 6 # CHECK: %[[batch_dataset:.*]], %[[batch_dataset_control:.*]] = tf_executor.island wraps "tf.BatchD…
|