Searched refs:input_buffer_shape (Results 1 – 1 of 1) sorted by relevance
77 TensorShape input_buffer_shape; in InitializeTensorList() local79 GetTensorListBufferShape(uninitialized_list, &input_buffer_shape)); in InitializeTensorList()80 if (input_buffer_shape.dim_size(0) != buffer_shape.dim_size(0)) { in InitializeTensorList()83 "input list size: ", input_buffer_shape.dim_size(0), in InitializeTensorList()