Searched refs:tensor_to_split_output (Results 1 – 1 of 1) sorted by relevance
755 absl::flat_hash_map<std::string, int> tensor_to_split_output; in CreateConcatAndSplitNodesForInputTensor() local764 if (tensor_to_split_output.contains(tensor_name)) continue; in CreateConcatAndSplitNodesForInputTensor()765 tensor_to_split_output[tensor_name] = concat_nodeouts.size(); in CreateConcatAndSplitNodesForInputTensor()773 const int num_tensors = tensor_to_split_output.size(); in CreateConcatAndSplitNodesForInputTensor()909 int output_index = tensor_to_split_output.at(tensor_name); in CreateConcatAndSplitNodesForInputTensor()931 absl::flat_hash_map<std::string, int> tensor_to_split_output; in CreateConcatAndSplitNodesForOutputTensor() local939 if (tensor_to_split_output.contains(tensor_name)) continue; in CreateConcatAndSplitNodesForOutputTensor()940 tensor_to_split_output[tensor_name] = concat_nodeouts.size(); in CreateConcatAndSplitNodesForOutputTensor()949 const int num_tensors = tensor_to_split_output.size(); in CreateConcatAndSplitNodesForOutputTensor()1045 int output_index = tensor_to_split_output.at(tensor_name); in CreateConcatAndSplitNodesForOutputTensor()