Searched refs:lt_workspace (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_maxpooling_op.cc126 mkl_context.lt_workspace)) /
151 dnnLayout_t lt_user_input = nullptr, lt_workspace = nullptr; member in struct:tensorflow::MklMaxPoolingOp::__anon26526
175 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&lt_workspace, prim_pooling_fwd,
186 CHECK_EQ(dnnLayoutDelete_F32(lt_workspace), E_SUCCESS);
421 dnnLayout_t lt_workspace; local
423 &lt_workspace, prim_pooling_fwd, dnnResourceWorkspace),
425 AllocTmpBuffer(context, &workspace_buf_tensor, lt_workspace,
429 dnnLayoutDelete_F32(lt_workspace);
H A Dmkl_avgpooling_op.cc98 AllocTmpBuffer(context, &workspace_tensor, mkl_context.lt_workspace,
157 lt_workspace = nullptr; member in struct:tensorflow::MklAvgPoolingOp::__anon26513
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::__anon26514
408 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&lt_workspace, prim_pooling_bwd,
421 CHECK_EQ(dnnLayoutDelete_F32(lt_workspace), E_SUCCESS);
H A Dmkl_lrn_op.cc507 dnnLayout_t lt_workspace = nullptr; member in struct:tensorflow::MklLRNGradOp::__anon26525
552 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(&lt_workspace, lrn_bwd,
723 dnnLayoutDelete_F32(lt_workspace);

Completed in 367 milliseconds