Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_lrn_op.cc190 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNOp
453 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNGradOp
H A Dmkl_pooling_ops_common.h657 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function
711 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function
H A Dmkl_slice_op.cc477 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklSliceOp
H A Dmkl_conv_grad_input_ops.cc562 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropInputOp
H A Dmkl_conv_grad_filter_ops.cc694 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropFilterOp
H A 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