Home
last modified time | relevance | path

Searched refs:second_tensor (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dmkl_util.h753 Tensor* second_tensor = nullptr; in AllocateOutputSetMklShape() local
758 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; variable
780 second_shape, &second_tensor));
782 second_tensor->flat<uint8>().data(),
783 second_tensor->flat<uint8>().size() * sizeof(uint8));