Searched defs:replicated_inputs (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 216 replicated_inputs, in ParseReplicateOpOperands()
277 replicated_inputs, in SetReplicateOpOperands()
327 replicated_inputs; in ParseReplicateOp() local
518 llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs, in BuildReplicateOp()
561 llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs, in build()
583 llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs, in build()
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | custom_training_loop_models_test.py | 101 def train_step(replicated_inputs): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_variable_runtime_reformatting.cc | 288 llvm::SmallVector<llvm::SmallVector<Value, 8>, 8> replicated_inputs; in AddInputsToReplicateOp() local
|
| D | tpu_cluster_formation.cc | 393 llvm::SmallVector<std::pair<ValueRange, Type>, 8> replicated_inputs; in ReplicateCluster() local
|