Searched defs:replicated_inputs (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_device.cc | 217 replicated_inputs, in ParseReplicateOpOperands()
279 replicated_inputs, in SetReplicateOpOperands()
331 replicated_inputs; in parse() local
437 llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs, in BuildReplicateOp()
573 llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs, in build()
595 llvm::ArrayRef<std::pair<ValueRange, Type>> replicated_inputs, in build()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | tpu_variable_runtime_reformatting.cc | 241 llvm::SmallVector<llvm::SmallVector<Value, 8>, 8> replicated_inputs; in AddInputsToReplicateOp() local
|
| H A D | tpu_cluster_formation.cc | 422 llvm::SmallVector<std::pair<ValueRange, Type>, 8> replicated_inputs; in ReplicateCluster() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_functional_ops.cc | 1934 std::vector<NodeDefBuilder::NodeOut> replicated_inputs; in ReplaceAndPartitionXLAShardingVariable() local
|