Searched refs:array_lifespans (Results 1 – 1 of 1) sorted by relevance
59 std::unordered_map<string, ArrayLifespan>* array_lifespans) { in UpdateArrayLifespan() argument60 if (array_lifespans->count(array_name)) { in UpdateArrayLifespan()61 auto& lifespan = array_lifespans->at(array_name); in UpdateArrayLifespan()70 (*array_lifespans)[array_name] = lifespan; in UpdateArrayLifespan()77 std::unordered_map<string, ArrayLifespan>* array_lifespans) { in ComputeArrayLifespans() argument78 CHECK(array_lifespans->empty()); in ComputeArrayLifespans()82 (*array_lifespans)[rnn_state.state_array()] = lifespan; in ComputeArrayLifespans()88 UpdateArrayLifespan(input, op_index, array_lifespans); in ComputeArrayLifespans()91 UpdateArrayLifespan(output, op_index, array_lifespans); in ComputeArrayLifespans()230 std::unordered_map<string, ArrayLifespan> array_lifespans; in AllocateTransientArrays() local[all …]