Searched refs:second_tensor (Results 1 – 2 of 2) sorted by relevance
743 Tensor* second_tensor = nullptr; in AllocateOutputSetMklShape() local748 second_shape, &second_tensor)); in AllocateOutputSetMklShape()750 second_tensor->flat<uint8>().data(), in AllocateOutputSetMklShape()751 second_tensor->flat<uint8>().size() * sizeof(uint8)); in AllocateOutputSetMklShape()765 Tensor* second_tensor = nullptr; variable770 second_shape, &second_tensor));772 second_tensor->flat<uint8>().data(),773 second_tensor->flat<uint8>().size() * sizeof(uint8));
986 Tensor* second_tensor = nullptr; in AllocatePersistentTensor() local998 &cached_filter_md_ptensor_, &second_tensor)); in AllocatePersistentTensor()999 *reinterpret_cast<memory::desc*>(second_tensor->flat<uint8>().data()) = in AllocatePersistentTensor()