Home
last modified time | relevance | path

Searched refs:needs_cast (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dnn_impl.py1438 needs_cast = x.dtype == dtypes.float16
1439 if needs_cast:
1479 if needs_cast:
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc558 bool needs_cast = float_type.getWidth() < 32; in matchAndRewrite() local
559 if (needs_cast) { in matchAndRewrite()
714 if (needs_cast) { in matchAndRewrite()