Searched refs:input_dims (Results 1 – 1 of 1) sorted by relevance
263 int64_t input_dims[] = {1, input->height, input->width, input->channels}; in allocate_input_tensor() local277 return TF_AllocateTensor(dt, input_dims, 4, in allocate_input_tensor()278 input_dims[1] * input_dims[2] * input_dims[3] * size); in allocate_input_tensor()