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.cc91 MklDnnShape output_mkl_shape; in ComputeScalar() local
137 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_dequantize_op.cc93 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_slice_op.cc386 MklDnnShape output_mkl_shape; in ComputeMklSlice() local
474 MklDnnShape* output_mkl_shape) { in AllocateOutputTensor()
Dmkl_quantize_op.cc269 MklDnnShape output_mkl_shape; in ComputeScalar() local
405 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_matmul_op_fused.cc137 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_pooling_ops_common.h548 MklDnnShape output_mkl_shape; in AllocateEmptyOutputTensor() local
651 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
703 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_lrn_op.cc197 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
459 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_softmax_op.cc277 MklDnnShape output_mkl_shape; in Compute() local
Dmkl_conv_ops.cc664 MklDnnShape output_mkl_shape; in Compute() local
852 MklDnnShape* output_mkl_shape, in AllocateOutputTensor()
1742 MklDnnShape* output_mkl_shape, in AllocateOutputTensor()
Dmkl_conv_grad_input_ops.cc570 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_conv_grad_filter_ops.cc685 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
Dmkl_matmul_ops_common.h397 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local