Searched refs:cast_to_f32 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tensor_tracer.py | 778 def _compute_signature(tensor, tf_op, cast_to_f32=True): argument
779 if cast_to_f32:
793 tsize = _compute_signature(tensor, array_ops.size, cast_to_f32=False)
798 def _show_max(tensor, cast_to_f32=True): argument
800 return _compute_signature(tensor, math_ops.reduce_max, cast_to_f32)
802 def _show_min(tensor, cast_to_f32=True): argument
804 return _compute_signature(tensor, math_ops.reduce_min, cast_to_f32)
806 def _show_norm(tensor, cast_to_f32=True): argument
808 return _compute_signature(tensor, linalg_ops.norm, cast_to_f32)
810 def _show_mean_and_variance(tensor, cast_to_f32=True): argument
[all …]
|