Searched refs:lt_workspace (Results 1 - 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
H A D | mkl_maxpooling_op.cc | 126 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(<_workspace, prim_pooling_fwd, 186 CHECK_EQ(dnnLayoutDelete_F32(lt_workspace), E_SUCCESS); 421 dnnLayout_t lt_workspace; local 423 <_workspace, prim_pooling_fwd, dnnResourceWorkspace), 425 AllocTmpBuffer(context, &workspace_buf_tensor, lt_workspace, 429 dnnLayoutDelete_F32(lt_workspace);
|
H A D | mkl_avgpooling_op.cc | 98 AllocTmpBuffer(context, &workspace_tensor, mkl_context.lt_workspace, 157 lt_workspace = nullptr; member in struct:tensorflow::MklAvgPoolingOp::__anon26513 195 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(<_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(<_workspace, prim_pooling_bwd, 421 CHECK_EQ(dnnLayoutDelete_F32(lt_workspace), E_SUCCESS);
|
H A D | mkl_lrn_op.cc | 507 dnnLayout_t lt_workspace = nullptr; member in struct:tensorflow::MklLRNGradOp::__anon26525 552 CHECK_EQ(dnnLayoutCreateFromPrimitive_F32(<_workspace, lrn_bwd, 723 dnnLayoutDelete_F32(lt_workspace);
|
Completed in 367 milliseconds