Home
last modified time | relevance | path

Searched refs:output_max_mkl_shape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_avgpooling_op.cc145 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
147 output_max_mkl_shape.SetMklTensor(false); in Compute()
151 output_max_mkl_shape); in Compute()
Dmkl_maxpooling_op.cc170 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
172 output_max_mkl_shape.SetMklTensor(false); in Compute()
176 output_max_mkl_shape); in Compute()
Dmkl_qmatmul_op.cc324 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
326 output_max_mkl_shape.SetMklTensor(false); in Compute()
330 output_max_mkl_shape); in Compute()
Dmkl_concat_op.cc784 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
786 output_max_mkl_shape.SetMklTensor(false); in Compute()
790 output_max_mkl_shape); in Compute()
Dmkl_conv_ops.cc1389 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
1391 output_max_mkl_shape.SetMklTensor(false); in Compute()
1400 output_max_mkl_shape); in Compute()
1418 output_max_mkl_shape); in Compute()
1428 output_max_mkl_shape); in Compute()