Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_reorder_replicate_and_partitioned_inputs.cc34 TF::TPUReplicatedInputOp replicated_input) { in ReorderReplicateAndPartitionedInputs()
106 getOperation()->walk([](TF::TPUReplicatedInputOp replicated_input) { in runOnOperation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_device.cc294 const auto& replicated_input = replicated_input_and_idx.value(); in SetReplicateOpOperands() local
448 for (auto& replicated_input : replicated_inputs) { in BuildReplicateOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc842 std::vector<NodeBuilder::NodeOut> replicated_input; in CreateConcatAndSplitNodesForInputTensor() local