Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_functional_ops.cc759 Node* split_vec_node = nullptr; in CreateConcatAndSplitNodesForInputTensor() local
775 .Finalize(graph, &split_vec_node)); in CreateConcatAndSplitNodesForInputTensor()
831 .Input(split_vec_node) in CreateConcatAndSplitNodesForInputTensor()
944 Node* split_vec_node = nullptr; in CreateConcatAndSplitNodesForOutputTensor() local
959 .Finalize(graph, &split_vec_node)); in CreateConcatAndSplitNodesForOutputTensor()
965 .Input(split_vec_node) in CreateConcatAndSplitNodesForOutputTensor()