Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_aggregate_ops.cc90 MklDnnShape output_mkl_shape; in ComputeScalar() local
136 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_dequantize_op.cc111 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_quantize_op.cc278 MklDnnShape output_mkl_shape; in ComputeScalar() local
414 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_slice_op.cc393 MklDnnShape output_mkl_shape; in ComputeMklSlice() local
482 MklDnnShape* output_mkl_shape) { in AllocateOutputTensor()
Dmkl_matmul_op_fused.cc151 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_pooling_ops_common.h563 MklDnnShape output_mkl_shape; in AllocateEmptyOutputTensor() local
666 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
718 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_lrn_op.cc198 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
460 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_softmax_op.cc300 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_conv_ops.cc824 MklDnnShape output_mkl_shape; in Compute() local
1088 MklDnnShape* output_mkl_shape, in AllocateOutputTensor()
2014 MklDnnShape* output_mkl_shape, in AllocateOutputTensor()
Dmkl_conv_grad_input_ops.cc573 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_conv_grad_filter_ops.cc705 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_matmul_ops_common.h470 MklDnnShape output_mkl_shape; variable