Searched refs:second_tensor (Results 1 – 1 of 1) sorted by relevance
753 Tensor* second_tensor = nullptr; in AllocateOutputSetMklShape() local758 second_shape, &second_tensor)); in AllocateOutputSetMklShape()760 second_tensor->flat<uint8>().data(), in AllocateOutputSetMklShape()761 second_tensor->flat<uint8>().size() * sizeof(uint8)); in AllocateOutputSetMklShape()775 Tensor* second_tensor = nullptr; variable780 second_shape, &second_tensor));782 second_tensor->flat<uint8>().data(),783 second_tensor->flat<uint8>().size() * sizeof(uint8));