Searched refs:map_root (Results 1 – 3 of 3) sorted by relevance
50 int map_root; member88 static void do_unshare(int map_root) in do_unshare() argument97 if (map_root) { in do_unshare()116 do_unshare(tc->map_root); in do_getxattr()
3693 auto* map_root = map_computation->root_instruction(); in HandleMap() local3694 if (map_root->opcode() == HloOpcode::kParameter) { in HandleMap()3696 map, map->mutable_operand(map_root->parameter_number())); in HandleMap()3699 if (map_root->opcode() == HloOpcode::kConstant) { in HandleMap()3700 if (!ShapeUtil::IsScalar(map_root->shape())) { in HandleMap()3703 auto clone = map_root->CloneWithNewOperands(map_root->shape(), {}); in HandleMap()3714 if (map_root->opcode() == HloOpcode::kFusion || !map_root->IsElementwise()) { in HandleMap()3718 for (auto* root_operand : map_root->operands()) { in HandleMap()3725 auto clone = map_root->CloneWithNewOperands(map->shape(), new_operands); in HandleMap()
1206 auto map_root = map_builder.AddInstruction( in TEST_F() local1238 auto& map_root_alloc = GetTopLevelAllocation(*assignment, map_root); in TEST_F()