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

  /external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc 580 MklDnnShape dnn_shape_diff_src; local
583 dnn_shape_diff_src.SetMklTensor(true);
585 dnn_shape_diff_src.SetMklLayout(&diff_src_pd);
586 dnn_shape_diff_src.SetElemType(MklDnnType<T>());
587 dnn_shape_diff_src.SetTfLayout(dnn_shape_src.GetDimension(),
592 dnn_shape_diff_src.SetMklTensor(false);
596 tf_shape_diff_src, dnn_shape_diff_src);
679 MklDnnShape dnn_shape_diff_src; local
680 dnn_shape_diff_src.SetMklTensor(false);
682 diff_dst_tensor.shape(), dnn_shape_diff_src);
746 MklDnnShape dnn_shape_diff_src; local
817 MklDnnShape dnn_shape_diff_src; local
    [all...]
mkl_fused_batch_norm_op.cc 1212 MklDnnShape dnn_shape_diff_src; variable
1296 MklDnnShape dnn_shape_diff_src; local
    [all...]

Completed in 82 milliseconds