Searched refs:_compute_signature (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): function
793 tsize = _compute_signature(tensor, array_ops.size, cast_to_f32=False)
800 return _compute_signature(tensor, math_ops.reduce_max, cast_to_f32)
804 return _compute_signature(tensor, math_ops.reduce_min, cast_to_f32)
808 return _compute_signature(tensor, linalg_ops.norm, cast_to_f32)
824 return _compute_signature(
|