Searched refs:concat_to_split (Results 1 – 1 of 1) sorted by relevance
891 const Edge* concat_to_split; in CreateConcatAndSplitNodesForInputTensor() local894 concat_to_split = edge; in CreateConcatAndSplitNodesForInputTensor()899 (*tpu_input_shapes)[concat_to_split].push_back( in CreateConcatAndSplitNodesForInputTensor()902 (*tpu_input_shapes)[concat_to_split].push_back( in CreateConcatAndSplitNodesForInputTensor()1023 const Edge* concat_to_split; in CreateConcatAndSplitNodesForOutputTensor() local1026 concat_to_split = edge; in CreateConcatAndSplitNodesForOutputTensor()1030 if (rank > 1) (*tpu_output_shapes)[concat_to_split].push_back(-1); in CreateConcatAndSplitNodesForOutputTensor()1032 (*tpu_output_shapes)[concat_to_split].push_back( in CreateConcatAndSplitNodesForOutputTensor()1034 (*tpu_output_shapes)[concat_to_split].push_back( in CreateConcatAndSplitNodesForOutputTensor()