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

  /external/tensorflow/tensorflow/core/kernels/
mkl_lrn_op.cc 247 TensorShape workspace_tf_shape; local
248 workspace_tf_shape.AddDim(0);
250 workspace_tf_shape, workspace_mkl_shape);
280 TensorShape workspace_tf_shape; local
281 workspace_tf_shape.AddDim(workspace_bytes);
283 workspace_tf_shape, workspace_mkl_shape);
mkl_maxpooling_op.cc 665 TensorShape workspace_tf_shape; local
666 workspace_tf_shape.AddDim(workspace_bytes);
668 &workspace_tensor, workspace_tf_shape,
    [all...]

Completed in 430 milliseconds