Searched refs:AllocTmpBuffer (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_grad_bias_ops.cc | 197 AllocTmpBuffer(context, mkl_tmp_outbackprop_buf, in MklPrepareConvolutionOutputs()
226 AllocTmpBuffer(context, mkl_tmp_input_buf, mkl_prim_internal_input, in MklPrepareConvolutionInputs()
|
| D | mkl_avgpooling_op.cc | 98 AllocTmpBuffer(context, &workspace_tensor, mkl_context.lt_workspace, in Compute()
191 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, lt_prim_input, in MklCreateLayoutsAndPrimitives()
283 AllocTmpBuffer(context, &outbackprop_buf_tensor, in Compute()
308 AllocTmpBuffer(context, &workspace_buf_tensor, mkl_context.lt_workspace, in Compute()
|
| D | mkl_maxpooling_op.cc | 374 AllocTmpBuffer(context, &input_buf_tensor, lt_input_prim, &input_buf); in MklCreatePrimitives()
383 AllocTmpBuffer(context, &outbackprop_buf_tensor, lt_outbackprop_prim, in MklCreatePrimitives()
427 AllocTmpBuffer(context, &workspace_buf_tensor, lt_workspace, in MklPrepareInputs()
434 AllocTmpBuffer(context, &tmp_output_buf_tensor, lt_outbackprop_prim, in MklPrepareInputs()
|
| D | mkl_conv_ops.cc | 760 AllocTmpBuffer(context, mkl_tmp_input_buf_tensor, mkl_lt_internal_input, in MklPrepareConvolutionInputs()
811 AllocTmpBuffer(context, mkl_tmp_bias_buf_tensor, mkl_lt_internal_bias, in MklPrepareConvolutionInputs()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 1040 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer() function
1050 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer() function
1065 inline void AllocTmpBuffer(OpKernelContext* context, Tensor* tensor_out, in AllocTmpBuffer() function
|