Searched refs:num_elements_in_tensor (Results 1 – 1 of 1) sorted by relevance
328 int num_elements_in_tensor = TensorTraits<T>::size(input_tensor); in ImportTensorData() local329 if (num_elements_in_tensor == input_flat_size) { in ImportTensorData()330 for (int i = 0; i < num_elements_in_tensor; i++) { in ImportTensorData()336 } else if (num_elements_in_tensor > 0 && in ImportTensorData()337 num_elements_in_tensor < input_flat_size) { in ImportTensorData()343 for (; i < num_elements_in_tensor; ++i) { in ImportTensorData()356 accessor_name, " (", num_elements_in_tensor, in ImportTensorData()