Searched refs:num_elements_in_tensor (Results 1 – 1 of 1) sorted by relevance
345 int num_elements_in_tensor = TensorTraits<T>::size(input_tensor); in ImportTensorData() local346 if (num_elements_in_tensor == input_flat_size) { in ImportTensorData()347 for (int i = 0; i < num_elements_in_tensor; i++) { in ImportTensorData()353 } else if (num_elements_in_tensor > 0 && in ImportTensorData()354 num_elements_in_tensor < input_flat_size) { in ImportTensorData()360 for (; i < num_elements_in_tensor; ++i) { in ImportTensorData()373 accessor_name, " (", num_elements_in_tensor, in ImportTensorData()