Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dmkl_concat_op.cc426 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
428 output_max_mkl_shape.SetMklTensor(false); in Compute()
432 output_max_mkl_shape); in Compute()
Dmkl_avgpooling_op.cc544 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
546 output_max_mkl_shape.SetMklTensor(false); in Compute()
550 output_max_mkl_shape); in Compute()
Dmkl_maxpooling_op.cc621 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
623 output_max_mkl_shape.SetMklTensor(false); in Compute()
627 output_max_mkl_shape); in Compute()
Dmkl_conv_ops.cc1566 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
1568 output_max_mkl_shape.SetMklTensor(false); in Compute()
1572 output_max_mkl_shape); in Compute()