Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_lrn_op.cc194 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNOp
444 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNGradOp
Dmkl_pooling_ops_common.h640 void AllocateOutputTensor( in AllocateOutputTensor() function
692 void AllocateOutputTensor( in AllocateOutputTensor() function
Dmkl_slice_op.cc443 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklSliceOp
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
Dmkl_conv_ops.cc1249 virtual void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvOp
1755 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklQuantizedConv2DSumReluOp