Searched refs:src_dtype (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 541 def _ConvertAndTest(template, src_dtype, dst_dtype): argument
543 x = c.Constant(np.array(template, dtype=src_dtype))
554 for src_dtype, dst_dtype in itertools.product(xla_types, xla_types):
555 _ConvertAndTest(x, src_dtype, dst_dtype)
568 def _ConvertAndTest(template, src_dtype, dst_dtype, dst_etype): argument
570 x = c.Constant(np.array(template, dtype=src_dtype))
574 expected = np.array(template, src_dtype).view(dst_dtype)
582 for src_dtype, dst_dtype in itertools.product(xla_types, xla_types):
583 _ConvertAndTest(x, src_dtype, dst_dtype, xla_types[dst_dtype])
|