HomeSort by relevance Sort by last modified time
    Searched refs:dnn_shape_diff_dst (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc 487 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; local
489 GetMklShape(context, diff_dst_index, &dnn_shape_diff_dst);
510 if (!dnn_shape_src.IsMklTensor() && !dnn_shape_diff_dst.IsMklTensor()) {
518 !dnn_shape_diff_dst.IsMklTensor()) {
533 dnn_shape_diff_dst.IsMklTensor()) {
535 diff_dst_md = dnn_shape_diff_dst.GetMklLayout();
538 dnn_shape_diff_dst.GetTfDataFormat();
549 diff_dst_md = dnn_shape_diff_dst.GetMklLayout();
560 if (dnn_shape_src.IsMklTensor() || dnn_shape_diff_dst.IsMklTensor()) {
676 MklDnnShape dnn_shape_diff_dst; local
743 MklDnnShape dnn_shape_diff_dst; local
814 MklDnnShape dnn_shape_diff_dst; local
    [all...]
mkl_fused_batch_norm_op.cc 1061 MklDnnShape dnn_shape_src, dnn_shape_diff_dst; variable
    [all...]

Completed in 93 milliseconds