Searched refs:add_instruction (Results 1 – 10 of 10) sorted by relevance
42 const HloInstruction* add_instruction = dot_instruction->operand(1); in TEST_F() local64 execution_profile.SetCyclesTakenBy(add_instruction, add_cycles); in TEST_F()75 add_instruction->name())))); in TEST_F()
89 add_instruction) { in Rsqrt()97 add_instruction) { in Mean()98 auto broadcast = add_instruction( in Mean()107 add_instruction) { in DynamicElementCountPerFeature()108 auto elements_per_feature_s32 = add_instruction( in DynamicElementCountPerFeature()116 add_instruction(HloInstruction::CreateGetDimensionSize( in DynamicElementCountPerFeature()118 elements_per_feature_s32 = add_instruction(HloInstruction::CreateBinary( in DynamicElementCountPerFeature()
166 add_instruction) { in GetExpandedFilterMask()180 auto mask1 = add_instruction(HloInstruction::CreateConstant( in GetExpandedFilterMask()182 auto broadcasted_mask1 = add_instruction(HloInstruction::CreateBroadcast( in GetExpandedFilterMask()184 auto mask2 = add_instruction(HloInstruction::CreateConstant( in GetExpandedFilterMask()186 auto broadcasted_mask2 = add_instruction(HloInstruction::CreateBroadcast( in GetExpandedFilterMask()193 return add_instruction(HloInstruction::CreateCompare( in GetExpandedFilterMask()
588 auto add_instruction = [&instructions_to_print, in ToString() local596 auto add_instructions_arround = [&add_instruction, &options](int index) { in ToString()600 add_instruction(i); in ToString()
58 void add_instruction(struct radeon_compiler *c, const char * inst_string);
458 void add_instruction(struct radeon_compiler *c, const char * inst_string) in add_instruction() function612 add_instruction(c, test->input[i]); in load_program()
264 *modified_add_function.add_instruction() = in ReplayAdaptedTransformations()
30 *message_.add_instruction() = instruction; in TransformationAddFunction()43 *message_.add_instruction() = instruction; in TransformationAddFunction()