Searched refs:second_tensor (Results 1 – 2 of 2) sorted by relevance
965 Tensor* second_tensor = nullptr; in AllocateOutputSetMklShape() local970 second_shape, &second_tensor)); in AllocateOutputSetMklShape()972 second_tensor->flat<uint8>().data(), in AllocateOutputSetMklShape()973 second_tensor->flat<uint8>().size() * sizeof(uint8)); in AllocateOutputSetMklShape()981 Tensor* second_tensor = nullptr; in AllocateOutputSetMklShape() local986 second_shape, &second_tensor)); in AllocateOutputSetMklShape()988 second_tensor->flat<uint8>().data(), in AllocateOutputSetMklShape()989 second_tensor->flat<uint8>().size() * sizeof(uint8)); in AllocateOutputSetMklShape()1000 Tensor* second_tensor = nullptr; in AllocateOutputSetMklShape() local1008 second_shape, &second_tensor)); in AllocateOutputSetMklShape()[all …]
1316 Tensor* second_tensor = nullptr; in AllocatePersistentTensor() local1323 &cached_filter_md_ptensor_, &second_tensor)); in AllocatePersistentTensor()1324 second_tensor->scalar<int32>()() = in AllocatePersistentTensor()