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

  /external/tensorflow/tensorflow/core/kernels/
mkl_softmax_op.cc 104 auto dst_pd = src.GetUsrMemPrimDesc();
mkl_aggregate_ops.cc 462 auto output_pd = dst.GetUsrMemPrimDesc();
mkl_concat_op.cc 705 auto src_mpd = srcs[k].GetUsrMemPrimDesc();
    [all...]
  /external/tensorflow/tensorflow/core/util/
mkl_util.h     [all...]

Completed in 146 milliseconds