Searched refs:num_output_elements (Results 1 – 9 of 9) sorted by relevance
53 int num_output_elements = 1; in ReshapeOutput() local61 num_output_elements *= value; in ReshapeOutput()65 output_shape->data[stretch_dim] = num_input_elements / num_output_elements; in ReshapeOutput()66 num_output_elements *= output_shape->data[stretch_dim]; in ReshapeOutput()70 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ReshapeOutput()
45 int num_output_elements; member91 op_data->num_output_elements = NumElements(output); in PrepareMax()
53 int num_output_elements = 1; in ResizeOutput() local61 num_output_elements *= value; in ResizeOutput()65 output_shape->data[stretch_dim] = num_input_elements / num_output_elements; in ResizeOutput()66 num_output_elements *= output_shape->data[stretch_dim]; in ResizeOutput()69 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ResizeOutput()
229 const size_t num_output_elements = in TestQS8() local248 std::vector<int8_t> output(num_output_elements); in TestQS8()249 std::vector<float> output_ref(num_output_elements); in TestQS8()367 const size_t num_output_elements = in TestF16() local386 std::vector<uint16_t> output(num_output_elements); in TestF16()387 std::vector<float> output_ref(num_output_elements); in TestF16()518 const size_t num_output_elements = in TestF32() local537 std::vector<float> output(num_output_elements); in TestF32()538 std::vector<float> output_ref(num_output_elements); in TestF32()563 const float output_min = num_output_elements == 1 ? in TestF32()[all …]
88 inline size_t num_output_elements() const { in num_output_elements() function141 std::vector<uint32_t> output(num_output_elements()); in TestX32()142 std::vector<uint32_t> output_ref(num_output_elements()); in TestX32()
44 int64 num_output_elements = 1; in ComputeBatchIndices() local52 for (int64 k = 0; k < (dim - 1) * num_output_elements; ++k) { in ComputeBatchIndices()53 (*out_indices)[num_output_elements + k] = (*out_indices)[k] + incr; in ComputeBatchIndices()55 num_output_elements *= dim; in ComputeBatchIndices()
479 const int64 num_output_elements = context->Value(num_elements_dim); in RegisterConstantNode() local480 data_size = max_bytes_per_data * num_output_elements; in RegisterConstantNode()929 const int64 num_output_elements = context->Value(num_elements_dim); in AppendNodeOutputParams() local930 data_size = max_bytes_per_data * num_output_elements; in AppendNodeOutputParams()933 CHECK_EQ(shapes.at(i).num_elements(), num_output_elements); in AppendNodeOutputParams()
820 const int num_output_elements = output_type.getNumElements(); in Verify() local822 if (num_output_elements % z_out != 0) { in Verify()828 if (num_input_elements / z_in != num_output_elements / z_out) { in Verify()
2365 // expected-error @+1 {{num_input_elements / z_in != num_output_elements / z_out}}