Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_aggregate_ops.cc90 MklDnnShape output_mkl_shape; in ComputeScalar() local
136 MklDnnShape output_mkl_shape; in Compute() local
H A 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
H A Dmkl_slice_op.cc393 MklDnnShape output_mkl_shape; in ComputeMklSlice() local
482 MklDnnShape* output_mkl_shape) { in AllocateOutputTensor()
H A Dmkl_quantize_op.cc278 MklDnnShape output_mkl_shape; in ComputeScalar() local
414 MklDnnShape output_mkl_shape; in Compute() local
H A Dmkl_lrn_op.cc198 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
460 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
H A Dmkl_dequantize_op.cc111 MklDnnShape output_mkl_shape; in Compute() local
H A Dmkl_matmul_op_fused.cc151 MklDnnShape output_mkl_shape; in Compute() local
H A 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()
H A Dmkl_softmax_op.cc300 MklDnnShape output_mkl_shape; in Compute() local
H A Dmkl_conv_grad_input_ops.cc573 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
H A Dmkl_conv_grad_filter_ops.cc705 MklDnnShape output_mkl_shape; in AllocateOutputTensor() local
H A Dmkl_matmul_ops_common.h470 MklDnnShape output_mkl_shape; variable