Searched refs:input_dims_count (Results 1 – 4 of 4) sorted by relevance
769 constexpr int input_dims_count = batch_size * input_size; in TF_LITE_MICRO_TEST() local771 float input_data[input_dims_count]; in TF_LITE_MICRO_TEST()800 constexpr int input_dims_count = batch_size * input_size; in TF_LITE_MICRO_TEST() local802 float input_data[input_dims_count]; in TF_LITE_MICRO_TEST()832 constexpr int input_dims_count = batch_size * input_size; in TF_LITE_MICRO_TEST() local845 int8_t input_quantized[input_dims_count]; in TF_LITE_MICRO_TEST()885 constexpr int input_dims_count = batch_size * input_size; in TF_LITE_MICRO_TEST() local898 int8_t input_quantized[input_dims_count]; in TF_LITE_MICRO_TEST()
369 const int input_dims_count = input_shape.DimensionsCount(); in Rfft2dHelper() local372 for (int i = 0; i < input_dims_count - 2; ++i) { in Rfft2dHelper()376 int input_height = input_dims_data[input_dims_count - 2]; in Rfft2dHelper()377 int input_width = input_dims_data[input_dims_count - 1]; in Rfft2dHelper()
447 const int input_dims_count = input_shape.DimensionsCount(); in EvalHybridImpl() local452 input_shape, input_dims_count - 1); in EvalHybridImpl()
88 const int input_dims_count = input_shape.DimensionsCount(); in FullyConnectedSparseWeight1x4Impl() local93 input_shape, input_dims_count - 1); in FullyConnectedSparseWeight1x4Impl()