Home
last modified time | relevance | path

Searched refs:AllocateOutputTensor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_lrn_op.cc151 AllocateOutputTensor(context, lrn_prim_desc, input_dims, input_format, in Compute()
194 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNOp
414 AllocateOutputTensor(context, lrn_bwd_prim_desc, orig_input_dims, in Compute()
444 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNGradOp
Dmkl_slice_op.cc419 AllocateOutputTensor(context, input_mkl_shape, &output_pd, size_dims, in ComputeMklSlice()
443 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklSliceOp
Dmkl_avgpooling_op.cc511 this->AllocateOutputTensor(context, *(pooling_fwd->GetPoolingFwdPd()), in Compute()
631 this->AllocateOutputTensor(context, *(pooling_bwd->GetPoolingBwdPd()), in Compute()
Dmkl_pooling_ops_common.h640 void AllocateOutputTensor( in AllocateOutputTensor() function
692 void AllocateOutputTensor( in AllocateOutputTensor() function
Dmkl_maxpooling_op.cc587 this->AllocateOutputTensor(context, *(pooling_fwd->GetPoolingFwdPd()), in Compute()
742 this->AllocateOutputTensor(context, *(pooling_bwd->GetPoolingBwdPd()), in Compute()
Dmkl_conv_ops.cc1085 AllocateOutputTensor(context, *conv_fwd_pd, dst_dims_mkl_order, tf_fmt, in Compute()
1249 virtual void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvOp
1755 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklQuantizedConv2DSumReluOp
1793 false>::AllocateOutputTensor(context, conv_prim_desc, in AllocateOutputTensor()
Dmkl_conv_grad_input_ops.cc533 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropInputOp
Dmkl_conv_grad_filter_ops.cc664 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropFilterOp