Home
last modified time | relevance | path

Searched refs:output_from_concat_node (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_functional_ops.cc829 Node* output_from_concat_node = nullptr; in CreateConcatAndSplitNodesForInputTensor() local
874 output_from_concat_node = tpu_replicated_input; in CreateConcatAndSplitNodesForInputTensor()
888 if (output_from_concat_node == nullptr) in CreateConcatAndSplitNodesForInputTensor()
889 output_from_concat_node = split_node; in CreateConcatAndSplitNodesForInputTensor()
893 if (edge->dst() == output_from_concat_node) { in CreateConcatAndSplitNodesForInputTensor()