Searched defs:num_output_elements (Results 1 – 10 of 10) sorted by relevance
228 const size_t num_output_elements = in TestQS8() local410 const size_t num_output_elements = in TestQU8() local585 const size_t num_output_elements = in TestF16() local806 const size_t num_output_elements = in TestF32() local
82 size_t num_output_elements; variable
43 size_t num_output_elements = NumElements(output_dims); in DepthToSpaceTest() local
87 inline size_t num_output_elements() const { in num_output_elements() function
186 const size_t num_output_elements = xnn_shape_multiply_all_dims(&output_value->shape); in xnn_define_static_reshape() local
61 int64_t non_zero_num_output_elements = 1, num_output_elements = 1; in ResizeOutput() local
44 int64_t num_output_elements = 1; in ComputeBatchIndices() local
186 const int num_output_elements = output_height * output_width; in upsample_bicubic2d_out_cuda_template() local
742 int64_t TensorIteratorBase::num_output_elements() const { in num_output_elements() function in at::TensorIteratorBase
1049 const int num_output_elements = output_type.getNumElements(); in verify() local