Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_lrn_op.cc189 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNOp
452 void AllocateOutputTensor( in AllocateOutputTensor() function in tensorflow::MklLRNGradOp
Dmkl_pooling_ops_common.h642 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function
696 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function
Dmkl_slice_op.cc469 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklSliceOp
Dmkl_conv_grad_input_ops.cc559 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropInputOp
Dmkl_conv_grad_filter_ops.cc674 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvCustomBackpropFilterOp
Dmkl_matmul_ops_common.h389 virtual void AllocateOutputTensor( in AllocateOutputTensor() function
Dmkl_conv_ops.cc848 virtual void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklConvOp
1738 void AllocateOutputTensor(OpKernelContext* context, in AllocateOutputTensor() function in tensorflow::MklQuantizedConv2DSumReluOp