Searched defs:AllocateOutputTensor (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | mkl_conv_grad_input_ops.cc | 443 AllocateOutputTensor(context, bwd_pd, bwd_output_dims, bwd_output_format, 453 void AllocateOutputTensor( function in class:tensorflow::MklConv2DCustomBackpropInputOp |
H A D | mkl_pooling_ops_common.h | 213 void AllocateOutputTensor( function in class:tensorflow::MklPoolingForwardOpBase 280 void AllocateOutputTensor( function in class:tensorflow::MklPoolingBackwardOpBase |
H A D | mkl_conv_grad_filter_ops.cc | 538 AllocateOutputTensor(context, bwd_pd, bwd_output_dims, bwd_output_format, 566 void AllocateOutputTensor( function in class:tensorflow::MklConv2DCustomBackpropFilterOp |
H A D | mkl_conv_ops.cc | 614 AllocateOutputTensor(context, conv_prim_desc, output_dims_mkl_order, 635 AllocateOutputTensor(context, conv_prim_desc, output_dims_mkl_order, 665 void AllocateOutputTensor( function in class:tensorflow::MklConv2DOp |
Completed in 78 milliseconds