| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | bitcast_convert_test.cc | 47 BitcastConvertType(a, S32); in TEST_F()
56 BitcastConvertType(a, F32); in TEST_F()
68 BitcastConvertType(a, F32); in TEST_F()
77 BitcastConvertType(a, F32); in XLA_TEST_F()
86 BitcastConvertType(a, S32); in TEST_F()
96 BitcastConvertType(a, F32); in TEST_F()
106 BitcastConvertType(param, S32); in TEST_F()
118 BitcastConvertType(param, F32); in TEST_F()
135 BitcastConvertType(reshape, F32); in TEST_F()
|
| D | convert_test.cc | 543 BitcastConvertType(all_bfloats_f32, U32); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 39 key[0] = BitcastConvertType(key[0], U32); in ThreeFry2x32()
40 key[1] = BitcastConvertType(key[1], U32); in ThreeFry2x32()
164 auto floats = BitcastConvertType(bits, F32); in StatelessRngUniformF32()
176 auto range = BitcastConvertType(maxval, unsigned_type) - in StatelessRngUniformInt()
177 BitcastConvertType(minval, unsigned_type); in StatelessRngUniformInt()
182 return minval + BitcastConvertType(dist_div_2, type) + in StatelessRngUniformInt()
183 BitcastConvertType(dist - dist_div_2, type); in StatelessRngUniformInt()
|
| D | comparators.cc | 84 auto signed_value = BitcastConvertType(value, signed_type); in BitcastConvertFloatingPointToIntegral()
85 auto unsigned_value = BitcastConvertType(value, unsigned_type); in BitcastConvertFloatingPointToIntegral()
87 BitcastConvertType(Sub(max_value, unsigned_value), signed_type); in BitcastConvertFloatingPointToIntegral()
|
| D | math.cc | 115 return Eq(BitcastConvertType(operand, U64), in IsNegZero()
118 return Eq(BitcastConvertType(operand, U32), in IsNegZero()
125 return Eq(BitcastConvertType(ConvertElementType(operand, F32), U32), in IsNegZero()
570 auto from_as_int = BitcastConvertType(from, int_type); in NextAfter()
571 auto to_as_int = BitcastConvertType(to, int_type); in NextAfter()
580 result_for_nan = BitcastConvertType(result_for_nan, int_type); in NextAfter()
638 return BitcastConvertType(result, shape.element_type()); in NextAfter()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 562 auto in_hi = xla::BitcastConvertType( in Compile()
566 auto bp_int = xla::BitcastConvertType(out_backprop, xla::U32); in Compile()
585 xla::BitcastConvertType(lhs, xla::S32), sixteen), in Compile()
589 xla::BitcastConvertType(rhs, xla::S32), sixteen), in Compile()
592 xla::Select(xla::Ge(xla::BitcastConvertType(lhs_criteria, xla::F32), in Compile()
593 xla::BitcastConvertType(rhs_criteria, xla::F32)), in Compile()
600 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_hi, xla::F32), in Compile()
603 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_lo, xla::F32), in Compile()
606 xla::ShiftLeft(xla::BitcastConvertType(pooled_hi, xla::U32), sixteen); in Compile()
608 xla::ShiftLeft(xla::BitcastConvertType(pooled_lo, xla::U32), sixteen), in Compile()
[all …]
|
| D | cast_op.cc | 78 input = xla::BitcastConvertType( in Compile()
80 xla::BitcastConvertType(input, same_width_int), in Compile()
129 output = xla::BitcastConvertType(input, dst_type_); in Compile()
|
| D | stateful_random_ops.cc | 137 return BitcastConvertType(x, xla::S32); in StatefulRngUniformFullInt()
145 return BitcastConvertType(x, xla::S64); in StatefulRngUniformFullInt()
217 auto counter = BitcastConvertType( in CompileImpl()
219 auto key = BitcastConvertType( in CompileImpl()
237 var = BitcastConvertType(var, state_element_type); in CompileImpl()
|
| D | stateless_random_ops.cc | 39 auto output = xla::BitcastConvertType(input, xla::U32) & in MaybeConvertF32ToBF16()
41 return xla::ConvertElementType(xla::BitcastConvertType(output, xla::F32), in MaybeConvertF32ToBF16()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 603 LocalOp ComputationBuilder::BitcastConvertType(const LocalOp& operand, in BitcastConvertType() function in xla::swig::ComputationBuilder
605 return xla::BitcastConvertType(operand.op(), new_element_type); in BitcastConvertType()
|
| D | local_computation_builder.h | 305 LocalOp BitcastConvertType(const LocalOp& operand,
|
| D | local_computation_builder.i | 317 %unignore xla::swig::ComputationBuilder::BitcastConvertType;
|
| D | xla_client.py | 1142 def BitcastConvertType(self, operand, new_element_type): member in ComputationBuilder
1152 return self._client.BitcastConvertType(operand, new_element_type)
|
| D | xla_client_test.py | 571 c.BitcastConvertType(x, dst_etype)
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1614 auto signed_value = BitcastConvertType(value, signed_type); in BitcastConvertFloatingPointToIntegral()
1615 auto unsigned_value = BitcastConvertType(value, unsigned_type); in BitcastConvertFloatingPointToIntegral()
1617 BitcastConvertType(Sub(max_value, unsigned_value), signed_type); in BitcastConvertFloatingPointToIntegral()
1719 XlaOp XlaBuilder::BitcastConvertType(const XlaOp& operand, in BitcastConvertType() function in xla::XlaBuilder
3353 XlaOp BitcastConvertType(const XlaOp& operand, PrimitiveType new_element_type) { in BitcastConvertType() function
3354 return operand.builder()->BitcastConvertType(operand, new_element_type); in BitcastConvertType()
|
| D | xla_builder.h | 505 XlaOp BitcastConvertType(const XlaOp& operand,
937 friend XlaOp BitcastConvertType(const XlaOp& operand,
1713 XlaOp BitcastConvertType(const XlaOp& operand, PrimitiveType new_element_type);
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 256 ## BitcastConvertType section in Operation Semantics
259 [`XlaBuilder::BitcastConvertType`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/x…
267 <b> `BitcastConvertType(operand, new_element_type)` </b>
|