Searched defs:tf_shape (Results 1 – 1 of 1) sorted by relevance
998 const TensorShape& tf_shape, in AllocateOutputSetMklShape()1019 const TensorShape& tf_shape, in AllocateOutputSetMklShape()1042 TensorShape tf_shape; in AllocTmpBuffer() local1052 TensorShape tf_shape; in AllocTmpBuffer() local1066 TensorShape tf_shape) { in AllocTmpBuffer()1091 TensorShape* tf_shape) { in MklSizesToTFSizes()