Searched refs:elements_per_feature_u32 (Results 1 – 1 of 1) sorted by relevance
116 auto elements_per_feature_u32 = add_instruction( in DynamicElementCountPerFeature() local126 elements_per_feature_u32 = add_instruction(HloInstruction::CreateBinary( in DynamicElementCountPerFeature()128 dynamic_dimension_size, elements_per_feature_u32)); in DynamicElementCountPerFeature()133 elements_per_feature_u32); in DynamicElementCountPerFeature()