Home
last modified time | relevance | path

Searched defs:AllocateOutputTensor (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_lrn_op.cc190 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNOp
453 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNGradOp
Dmkl_pooling_ops_common.h657 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function
711 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function
Dmkl_slice_op.cc477 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklSliceOp
Dmkl_conv_grad_input_ops.cc562 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropInputOp
Dmkl_conv_grad_filter_ops.cc694 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropFilterOp
Dmkl_conv_ops.cc1084 virtual void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvOp
2010 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklQuantizedConv2DSumReluOp