Searched refs:num_input_elements (Results 1 – 8 of 8) sorted by relevance
42 int num_input_elements = NumElements(input); in ReshapeOutput() local65 output_shape->data[stretch_dim] = num_input_elements / num_output_elements; in ReshapeOutput()70 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ReshapeOutput()
51 int num_input_elements = NumElements(input); in ResizeOutput() local65 output_shape->data[stretch_dim] = num_input_elements / num_output_elements; in ResizeOutput()69 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ResizeOutput()
45 int64 num_input_elements = 1; in ComputeBatchIndices() local51 const int64 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<uint32_t> input(XNN_EXTRA_BYTES / sizeof(uint32_t) + num_input_elements()); in TestX32()
1614 int32_t num_input_elements = 1; in VisitFullyConnectedNode() local1622 num_input_elements *= input_tensor.dims->data[i]; in VisitFullyConnectedNode()1642 if (num_input_elements % input_channels != 0) { in VisitFullyConnectedNode()1654 if (output_tensor.dims->data[0] != num_input_elements / input_channels) { in VisitFullyConnectedNode()1660 num_input_elements / input_channels, node->inputs->data[0]); in VisitFullyConnectedNode()
804 const int num_input_elements = input_type.getNumElements(); in Verify() local805 if (num_input_elements % z_in != 0) { in Verify()828 if (num_input_elements / z_in != num_output_elements / z_out) { in Verify()
459 const size_t num_input_elements = product_all_dims(&values[node->inputs[0]].shape); in xnn_create_runtime_v2() local476 runtime->opdata[i].batch_size = num_input_elements / input_channels; in xnn_create_runtime_v2()
2365 // expected-error @+1 {{num_input_elements / z_in != num_output_elements / z_out}}