Searched refs:output_num_elements (Results 1 – 2 of 2) sorted by relevance
325 const int64_t output_num_elements = output_num_pixels * channels(); in TestNCHWxF32() local338 …output_ref[batch_index * output_num_elements + c * output_num_pixels + output_y * output_width() +… in TestNCHWxF32()378 … ASSERT_NEAR(output[i * output_num_elements + c * output_num_pixels + y * output_width() + x], in TestNCHWxF32()379 … output_ref[i * output_num_elements + c * output_num_pixels + y * output_width() + x], in TestNCHWxF32()
2773 const int output_num_elements = in IsReductionCandidateForSimplification() local2775 *is_single_element_op = input_num_elements == 1 && output_num_elements == 1; in IsReductionCandidateForSimplification()