| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_lrn_op.cc | 156 AllocateOutputTensor(context, lrn_prim_desc, input_dims, input_format, in Compute()
189 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNOp
412 AllocateOutputTensor(context, lrn_bwd_prim_desc, orig_input_dims, in Compute()
452 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNGradOp
|
| D | mkl_avgpooling_op.cc | 122 this->AllocateOutputTensor(context, *(pooling_fwd->GetPoolingFwdPd()), in Compute()
255 this->AllocateOutputTensor(context, *(pooling_bwd->GetPoolingBwdPd()), in Compute()
|
| D | mkl_maxpooling_op.cc | 146 this->AllocateOutputTensor(context, *(pooling_fwd->GetPoolingFwdPd()), in Compute()
310 this->AllocateOutputTensor(context, *(pooling_bwd->GetPoolingBwdPd()), in Compute()
|
| D | mkl_slice_op.cc | 443 AllocateOutputTensor(context, input_mkl_shape, &output_pd, size_dims, in ComputeMklSlice()
469 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklSliceOp
|
| D | mkl_pooling_ops_common.h | 642 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function
696 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function
|
| D | mkl_conv_ops.cc | 666 AllocateOutputTensor(context, *conv_fwd_pd, dst_dims_mkl_order, tf_fmt, in Compute()
848 virtual void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvOp
1738 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklQuantizedConv2DSumReluOp
1776 false>::AllocateOutputTensor(context, conv_prim_desc, in AllocateOutputTensor()
|
| D | mkl_conv_grad_input_ops.cc | 559 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropInputOp
|
| D | mkl_qmatmul_op.cc | 241 this->AllocateOutputTensor(context, *matmul_fwd_pd, dst_dims_mkl_order, in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 674 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropFilterOp
|
| D | mkl_matmul_ops_common.h | 389 virtual void AllocateOutputTensor( in AllocateOutputTensor() function
|