Home
last modified time | relevance | path

Searched defs:output_mkl_shape (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_softmax_op.cc137 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_aggregate_ops.cc214 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_slice_op.cc378 MklDnnShape output_mkl_shape; in ComputeMklSlice() local
448 MklDnnShape* output_mkl_shape) { in AllocateOutputTensor()
Dmkl_conv_grad_bias_ops.cc92 MklShape output_mkl_shape; in Compute() local
Dmkl_pooling_ops_common.h558 MklDnnShape output_mkl_shape; in AllocateEmptyOutputTensor() local
648 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
700 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_lrn_op.cc202 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
452 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_conv_grad_input_ops.cc544 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_conv_grad_filter_ops.cc677 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_conv_ops.cc1264 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local