Searched refs:output_num_elements (Results 1 – 3 of 3) sorted by relevance
37 int output_num_elements = output_shape.FlatSize(); in BroadcastArgs() local38 for (int i = 0; i < output_num_elements; ++i) { in BroadcastArgs()39 int backward_i = output_num_elements - 1 - i; in BroadcastArgs()
579 const int64_t output_num_elements = output_num_pixels * channels(); in TestNCHWxF32() local592 …output_ref[batch_index * output_num_elements + c * output_num_pixels + output_y * output_width() +… in TestNCHWxF32()632 … ASSERT_NEAR(output[i * output_num_elements + c * output_num_pixels + y * output_width() + x], in TestNCHWxF32()633 … output_ref[i * output_num_elements + c * output_num_pixels + y * output_width() + x], in TestNCHWxF32()
2823 const int output_num_elements = in IsReductionCandidateForSimplification() local2825 *is_single_element_op = input_num_elements == 1 && output_num_elements == 1; in IsReductionCandidateForSimplification()