Searched refs:AllocateWorkspaceTensor (Results 1 – 2 of 2) sorted by relevance
187 AllocateWorkspaceTensor(context, *(pooling_fwd->GetPoolingFwdPd()), in Compute()208 void AllocateWorkspaceTensor( in AllocateWorkspaceTensor() function in tensorflow::MklMaxPoolingOp
163 AllocateWorkspaceTensor(context, lrn_prim_desc, &workspace_dnn_data); in Compute()262 void AllocateWorkspaceTensor( in AllocateWorkspaceTensor() function in tensorflow::MklLRNOp