Searched refs:unsigned_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | comparators.cc | 41 PrimitiveType unsigned_type; in BitcastConvertFloatingPointToIntegral() local
49 unsigned_type = U16; in BitcastConvertFloatingPointToIntegral()
56 unsigned_type = U32; in BitcastConvertFloatingPointToIntegral()
63 unsigned_type = U64; in BitcastConvertFloatingPointToIntegral()
85 auto unsigned_value = BitcastConvertType(value, unsigned_type); in BitcastConvertFloatingPointToIntegral()
|
| D | prng.cc | 174 PrimitiveType type, PrimitiveType unsigned_type) { in StatelessRngUniformInt() argument
176 auto range = BitcastConvertType(maxval, unsigned_type) - in StatelessRngUniformInt()
177 BitcastConvertType(minval, unsigned_type); in StatelessRngUniformInt()
180 ShiftRightLogical(dist, ConstantR0WithType(builder, unsigned_type, 1)); in StatelessRngUniformInt()
|
| D | prng.h | 50 PrimitiveType type, PrimitiveType unsigned_type);
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | gather_op_test.py | 166 for unsigned_type in (dtypes.uint32, dtypes.uint64):
168 np.array([[1, 2, 3], [7, 8, 9]]), unsigned_type)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1584 PrimitiveType unsigned_type; in BitcastConvertFloatingPointToIntegral() local
1592 unsigned_type = U16; in BitcastConvertFloatingPointToIntegral()
1599 unsigned_type = U32; in BitcastConvertFloatingPointToIntegral()
1606 unsigned_type = U64; in BitcastConvertFloatingPointToIntegral()
1615 auto unsigned_value = BitcastConvertType(value, unsigned_type); in BitcastConvertFloatingPointToIntegral()
|