Home
last modified time | relevance | path

Searched refs:final_dim_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dspace_to_batch_nd.cc74 int final_dim_size = (input_size->data[dim + 1] + paddings_data[dim * 2] + in ResizeOutputTensor() local
76 TF_LITE_ENSURE_EQ(context, final_dim_size % block_shape[dim], 0); in ResizeOutputTensor()
77 output_size->data[dim + 1] = final_dim_size / block_shape[dim]; in ResizeOutputTensor()