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

  /external/tensorflow/tensorflow/core/kernels/
mkl_relu_op.cc 584 auto diff_src_pd = relu_bwd_pd.diff_src_primitive_desc(); local
585 dnn_shape_diff_src.SetMklLayout(&diff_src_pd);
590 tf_shape_diff_src.AddDim(diff_src_pd.get_size() / sizeof(T));
    [all...]
mkl_fused_batch_norm_op.cc 1216 auto diff_src_pd = bnrm_fwd_pd.dst_primitive_desc(); variable
    [all...]

Completed in 173 milliseconds