Searched refs:dynamic_output_mapping (Results 1 – 1 of 1) sorted by relevance
1437 ShapeTree<absl::flat_hash_map<int64_t, int64_t>> dynamic_output_mapping( in HandleConditional() local1536 if (dynamic_output_mapping.element(index).contains(i)) { in HandleConditional()1539 dynamic_output_mapping.mutable_element(index)->emplace( in HandleConditional()1556 if (dynamic_output_mapping.element(index).contains(i)) { in HandleConditional()1598 dynamic_output_mapping.ForEachElement( in HandleConditional()1692 ShapeTree<absl::flat_hash_map<int64_t, int64_t>> dynamic_output_mapping( in HandleWhile() local1701 dynamic_output_mapping.mutable_element(index)->emplace(dim, in HandleWhile()1713 if (!dynamic_output_mapping.mutable_element(index)->contains(dim)) { in HandleWhile()1720 dynamic_output_mapping.mutable_element(index)->emplace( in HandleWhile()1744 TF_RETURN_IF_ERROR(dynamic_output_mapping.ForEachElementWithStatus( in HandleWhile()[all …]