Home
last modified time | relevance | path

Searched refs:output_min_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
427 output_min_mkl_shape.SetMklTensor(false); in Compute()
430 output_min_mkl_shape); in Compute()
Dmkl_avgpooling_op.cc544 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
545 output_min_mkl_shape.SetMklTensor(false); in Compute()
548 output_min_mkl_shape); in Compute()
Dmkl_maxpooling_op.cc621 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
622 output_min_mkl_shape.SetMklTensor(false); in Compute()
625 output_min_mkl_shape); in Compute()
Dmkl_conv_ops.cc1566 MklDnnShape output_min_mkl_shape, output_max_mkl_shape; in Compute() local
1567 output_min_mkl_shape.SetMklTensor(false); in Compute()
1570 output_min_mkl_shape); in Compute()