| /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/tf2xla/kernels/ |
| D | pooling_ops.cc | 586 auto in_hi = xla::BitcastConvertType( in Compile()
589 auto bp_int = xla::BitcastConvertType(out_backprop, xla::U32); in Compile()
608 xla::BitcastConvertType(lhs, xla::S32), sixteen), in Compile()
612 xla::BitcastConvertType(rhs, xla::S32), sixteen), in Compile()
615 xla::Select(xla::Ge(xla::BitcastConvertType(lhs_criteria, xla::F32), in Compile()
616 xla::BitcastConvertType(rhs_criteria, xla::F32)), in Compile()
623 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_hi, xla::F32), in Compile()
626 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_lo, xla::F32), in Compile()
629 xla::ShiftLeft(xla::BitcastConvertType(pooled_hi, xla::U32), sixteen); in Compile()
631 xla::ShiftLeft(xla::BitcastConvertType(pooled_lo, xla::U32), sixteen), in Compile()
[all …]
|
| D | cast_op.cc | 83 input = xla::BitcastConvertType( in Compile()
85 xla::BitcastConvertType(input, same_width_int), in Compile()
158 input = xla::BitcastConvertType(status_or_input.ConsumeValueOrDie(), in Compile()
198 output = xla::BitcastConvertType(input, dst_type_); in Compile()
|
| D | stateful_random_ops.cc | 103 output.value = BitcastConvertType(output.value, type); in StatefulRngUniformFullInt()
142 auto state = BitcastConvertType( in StateAndKeyFromVariable()
144 auto key = BitcastConvertType( in StateAndKeyFromVariable()
151 BitcastConvertType(xla::Slice(var, {0}, {kStateSize}, {1}), xla::U64); in StateAndKeyFromVariable()
153 BitcastConvertType(xla::Slice(var, {kStateSize}, {kStateSize + 1}, {1}), in StateAndKeyFromVariable()
210 var = BitcastConvertType(var, state_element_type); in CompileImpl()
463 delta = BitcastConvertType(delta, xla::U64); in Compile()
469 var = BitcastConvertType(var, state_element_type); in Compile()
|
| D | stateless_random_ops.cc | 69 xla::XlaOp output = xla::BitcastConvertType(input, xla::U32) & in MaybeConvertF32ToBF16()
71 return xla::ConvertElementType(xla::BitcastConvertType(output, xla::F32), in MaybeConvertF32ToBF16()
132 return BitcastConvertType(output.value, type); in StatelessRngUniformFullInt()
|
| D | stateless_random_ops_v2.cc | 72 key = BitcastConvertType(key, xla::U64); in BitGenerator()
73 counter = BitcastConvertType(counter, xla::U64); in BitGenerator()
77 new_counter = BitcastConvertType(new_counter, xla::S64); in BitGenerator()
158 return xla::RngOutput{BitcastConvertType(output.value, type), in StatelessRngUniformFullInt()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | topk_ops.cc | 49 xla::XlaOp input_sm32 = xla::BitcastConvertType(input, xla::S32); in CreateKthOrderStatisticComputation()
92 return xla::BitcastConvertType(top_r1, xla::F32); in CreateKthOrderStatisticComputation()
194 xla::XlaOp topk_r2_s32 = xla::BitcastConvertType(topk_r2, xla::S32); in CreateTopKUnique()
295 xla::XlaOp input_u32_r2 = xla::BitcastConvertType(input, xla::U32); in CreateMakeUnique()
306 return xla::BitcastConvertType(or_r2, xla::F32); in CreateMakeUnique()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 53 key[0] = BitcastConvertType(key[0], U32); in ThreeFry2x32()
54 key[1] = BitcastConvertType(key[1], U32); in ThreeFry2x32()
503 XlaOp range = BitcastConvertType(maxval, unsigned_type) - in ConvertRandomBitsToUniformInt()
504 BitcastConvertType(minval, unsigned_type); in ConvertRandomBitsToUniformInt()
509 return minval + BitcastConvertType(dist_div_2, type) + in ConvertRandomBitsToUniformInt()
510 BitcastConvertType(dist - dist_div_2, type); in ConvertRandomBitsToUniformInt()
|
| D | math.cc | 147 return Eq(BitcastConvertType(operand, U64), in IsNegZero()
150 return Eq(BitcastConvertType(operand, U32), in IsNegZero()
157 return Eq(BitcastConvertType(ConvertElementType(operand, F32), U32), in IsNegZero()
1363 auto from_as_int = BitcastConvertType(from, int_type); in NextAfter()
1364 auto to_as_int = BitcastConvertType(to, int_type); in NextAfter()
1373 result_for_nan = BitcastConvertType(result_for_nan, int_type); in NextAfter()
1431 return BitcastConvertType(result, shape.element_type()); in NextAfter()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 86 ops.def("BitcastConvertType", &BitcastConvertType, py::arg("operand"), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 628 ops.BitcastConvertType(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 788 XlaOp BitcastConvertType(XlaOp operand, PrimitiveType new_element_type);
1359 friend XlaOp BitcastConvertType(XlaOp operand,
2355 XlaOp BitcastConvertType(XlaOp operand, PrimitiveType new_element_type);
|
| D | xla_builder.cc | 2100 XlaOp XlaBuilder::BitcastConvertType(XlaOp operand, in BitcastConvertType() function in xla::XlaBuilder
4802 XlaOp BitcastConvertType(const XlaOp operand, PrimitiveType new_element_type) { in BitcastConvertType() function
4803 return operand.builder()->BitcastConvertType(operand, new_element_type); in BitcastConvertType()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 656 // TODO(mgester): Due to restrictions of xla::BitcastConvertType we currently
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 599 value_map[op] = xla::BitcastConvertType( in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 342 ## BitcastConvertType section in Operation Semantics
345 [`XlaBuilder::BitcastConvertType`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/x…
353 <b> `BitcastConvertType(operand, new_element_type)` </b>
|