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

  /external/tensorflow/tensorflow/core/kernels/
mkl_maxpooling_op.cc 127 mkl_context.lt_workspace)) /
152 dnnLayout_t lt_user_input = nullptr, lt_workspace = nullptr; member in struct:tensorflow::MklMaxPoolingOp::__anon45427
176 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&lt_workspace, prim_pooling_fwd,
187 CHECK_EQ(dnnLayoutDelete_F32(lt_workspace), E_SUCCESS);
423 dnnLayout_t lt_workspace; local
425 &lt_workspace, prim_pooling_fwd, dnnResourceWorkspace),
427 AllocTmpBuffer(context, &workspace_buf_tensor, lt_workspace,
431 dnnLayoutDelete_F32(lt_workspace);
    [all...]
mkl_avgpooling_op.cc 98 AllocTmpBuffer(context, &workspace_tensor, mkl_context.lt_workspace,
157 lt_workspace = nullptr; member in struct:tensorflow::MklAvgPoolingOp::__anon45422
195 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&lt_workspace, prim_pooling_fwd,
308 AllocTmpBuffer(context, &workspace_buf_tensor, mkl_context.lt_workspace,
350 lt_prim_outbackprop = nullptr, lt_workspace = nullptr; member in struct:tensorflow::MklAvgPoolingGradOp::__anon45423
408 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&lt_workspace, prim_pooling_bwd,
421 CHECK_EQ(dnnLayoutDelete_F32(lt_workspace), E_SUCCESS);

Completed in 70 milliseconds