Searched refs:tensor_to_split_output (Results 1 – 1 of 1) sorted by relevance
705 absl::flat_hash_map<std::string, int> tensor_to_split_output; in CreateConcatAndSplitNodesForInputTensor() local714 if (tensor_to_split_output.contains(tensor_name)) continue; in CreateConcatAndSplitNodesForInputTensor()715 tensor_to_split_output[tensor_name] = concat_nodeouts.size(); in CreateConcatAndSplitNodesForInputTensor()723 const int num_tensors = tensor_to_split_output.size(); in CreateConcatAndSplitNodesForInputTensor()859 int output_index = tensor_to_split_output.at(tensor_name); in CreateConcatAndSplitNodesForInputTensor()881 absl::flat_hash_map<std::string, int> tensor_to_split_output; in CreateConcatAndSplitNodesForOutputTensor() local889 if (tensor_to_split_output.contains(tensor_name)) continue; in CreateConcatAndSplitNodesForOutputTensor()890 tensor_to_split_output[tensor_name] = concat_nodeouts.size(); in CreateConcatAndSplitNodesForOutputTensor()899 const int num_tensors = tensor_to_split_output.size(); in CreateConcatAndSplitNodesForOutputTensor()995 int output_index = tensor_to_split_output.at(tensor_name); in CreateConcatAndSplitNodesForOutputTensor()