HomeSort by relevance Sort by last modified time
    Searched refs:AllocateWorkspaceTensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc 158 AllocateWorkspaceTensor(context, lrn_prim_desc, &workspace_dnn_data);
267 void AllocateWorkspaceTensor(
mkl_maxpooling_op.cc 632 AllocateWorkspaceTensor(context, *(pooling_fwd->GetPoolingFwdPd()),
654 void AllocateWorkspaceTensor(
    [all...]

Completed in 73 milliseconds