Home
last modified time | relevance | path

Searched defs:replicated_input (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_reorder_replicate_and_partitioned_inputs.cc34 TF::TPUReplicatedInputOp replicated_input) { in ReorderReplicateAndPartitionedInputs()
110 getFunction()->walk([](TF::TPUReplicatedInputOp replicated_input) { in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_device.cc296 const auto& replicated_input = replicated_input_and_idx.value(); in SetReplicateOpOperands() local
533 for (auto& replicated_input : replicated_inputs) { in BuildReplicateOp() local
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc792 std::vector<NodeBuilder::NodeOut> replicated_input; in CreateConcatAndSplitNodesForInputTensor() local