Home
last modified time | relevance | path

Searched defs:dnn_shape_src (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_relu_op.cc445 MklDnnShape dnn_shape_src; in Compute() local
600 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; in Compute() local
777 MklDnnShape dnn_shape_src; in Compute_Scalar() local
844 MklDnnShape dnn_shape_src; in Compute_Scalar() local
925 MklDnnShape dnn_shape_src; in Compute_Scalar() local
1010 MklDnnShape dnn_shape_src; in Compute_Scalar() local
1086 MklDnnShape dnn_shape_src; in Compute_Scalar() local
H A Dmkl_fused_batch_norm_op.cc716 MklDnnShape dnn_shape_src; in Compute() local
1142 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; in Compute() local