Searched refs:in_dtype (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1454 def _CreateConstantComputation(self, in_dtype, out_dtype): argument
1457 in_dtype.__name__, out_dtype.__name__))
1459 xla_client.shape_from_pyval(np.array(0, dtype=in_dtype)))
1516 "testcase_name": "_{}_{}".format(in_dtype.__name__, out_dtype.__name__),
1517 "in_dtype": in_dtype,
1519 } for in_dtype, out_dtype in [[np.float32, np.int32]])
1520 def testMapEachElementToConstant(self, in_dtype, out_dtype): argument
1523 [ops.Constant(c, np.array([1.0, 2.0, 3.0, 4.0], dtype=in_dtype))],
1524 self._CreateConstantComputation(in_dtype, out_dtype), [0])
|