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

  /external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc 245 MklDnnShape workspace_mkl_shape; local
246 workspace_mkl_shape.SetMklTensor(false);
250 workspace_tf_shape, workspace_mkl_shape);
276 MklDnnShape workspace_mkl_shape; local
279 workspace_mkl_shape.SetMklTensor(false);
283 workspace_tf_shape, workspace_mkl_shape);
mkl_maxpooling_op.cc 663 MklDnnShape workspace_mkl_shape; local
664 workspace_mkl_shape.SetMklTensor(false);
669 workspace_mkl_shape);
    [all...]

Completed in 375 milliseconds