Searched refs:num_input_elements (Results 1 – 7 of 7) sorted by relevance
51 int64_t non_zero_num_input_elements = 1, num_input_elements = 1; in ResizeOutput() local55 num_input_elements *= value; in ResizeOutput()76 if (num_input_elements == 0 && num_output_elements != 0) { in ResizeOutput()85 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ResizeOutput()
45 int64_t num_input_elements = 1; in ComputeBatchIndices() local51 const int64_t incr = bcast[i] > 1 ? 0 : num_input_elements; in ComputeBatchIndices()56 num_input_elements *= reshape[i]; in ComputeBatchIndices()
43 inline size_t num_input_elements() const { in num_input_elements() function140 std::vector<uint8_t> input(XNN_EXTRA_BYTES / sizeof(uint8_t) + num_input_elements()); in TestX8()255 std::vector<uint16_t> input(XNN_EXTRA_BYTES / sizeof(uint16_t) + num_input_elements()); in TestX16()370 std::vector<uint32_t> input(XNN_EXTRA_BYTES / sizeof(uint32_t) + num_input_elements()); in TestX32()
36 const size_t num_input_elements = xnn_shape_multiply_all_dims(&values[node->inputs[0]].shape); in create_fully_connected_operator() local131 opdata->batch_size = num_input_elements / input_channels; in create_fully_connected_operator()
2138 int32_t num_input_elements = 1; in VisitFullyConnectedNode() local2146 num_input_elements *= input_tensor.dims->data[i]; in VisitFullyConnectedNode()2166 if (num_input_elements % input_channels != 0) { in VisitFullyConnectedNode()2178 if (output_tensor.dims->data[0] != num_input_elements / input_channels) { in VisitFullyConnectedNode()2184 num_input_elements / input_channels, node->inputs->data[0]); in VisitFullyConnectedNode()
891 const int num_input_elements = input_type.getNumElements(); in Verify() local892 if (num_input_elements % z_in != 0) { in Verify()915 if (num_input_elements / z_in != num_output_elements / z_out) { in Verify()
2413 // expected-error @+1 {{num_input_elements / z_in != num_output_elements / z_out}}