| /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 | 638 auto in_hi = xla::BitcastConvertType( in Compile()
641 auto bp_int = xla::BitcastConvertType(out_backprop, xla::U32); in Compile()
660 xla::BitcastConvertType(lhs, xla::S32), sixteen), in Compile()
664 xla::BitcastConvertType(rhs, xla::S32), sixteen), in Compile()
667 xla::Select(xla::Ge(xla::BitcastConvertType(lhs_criteria, xla::F32), in Compile()
668 xla::BitcastConvertType(rhs_criteria, xla::F32)), in Compile()
675 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_hi, xla::F32), in Compile()
678 xla::ReduceWindow(xla::BitcastConvertType(in_hi_bp_lo, xla::F32), in Compile()
681 xla::ShiftLeft(xla::BitcastConvertType(pooled_hi, xla::U32), sixteen); in Compile()
683 xla::ShiftLeft(xla::BitcastConvertType(pooled_lo, xla::U32), sixteen), in Compile()
[all …]
|
| D | stateful_random_ops.cc | 104 output.value = BitcastConvertType(output.value, type); in StatefulRngUniformFullInt()
143 auto state = BitcastConvertType( in StateAndKeyFromVariable()
145 auto key = BitcastConvertType( in StateAndKeyFromVariable()
152 BitcastConvertType(xla::Slice(var, {0}, {kStateSize}, {1}), xla::U64); in StateAndKeyFromVariable()
154 BitcastConvertType(xla::Slice(var, {kStateSize}, {kStateSize + 1}, {1}), in StateAndKeyFromVariable()
211 var = BitcastConvertType(var, state_element_type); in CompileImpl()
464 delta = BitcastConvertType(delta, xla::U64); in Compile()
470 var = BitcastConvertType(var, state_element_type); in Compile()
|
| D | cast_op.cc | 83 input = xla::BitcastConvertType( in Compile()
85 xla::BitcastConvertType(input, same_width_int), in Compile()
137 output = xla::BitcastConvertType(input, dst_type_); in Compile()
|
| D | stateless_random_ops_v2.cc | 74 key = BitcastConvertType(key, xla::U64); in BitGenerator()
75 counter = BitcastConvertType(counter, xla::U64); in BitGenerator()
79 new_counter = BitcastConvertType(new_counter, xla::S64); in BitGenerator()
160 return xla::RngOutput{BitcastConvertType(output.value, type), in StatelessRngUniformFullInt()
|
| D | stateless_random_ops.cc | 70 xla::XlaOp output = xla::BitcastConvertType(input, xla::U32) & in MaybeConvertF32ToBF16()
72 return xla::ConvertElementType(xla::BitcastConvertType(output, xla::F32), in MaybeConvertF32ToBF16()
133 return BitcastConvertType(output.value, type); in StatelessRngUniformFullInt()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | bitcast_dtypes_expander.cc | 81 input = BitcastConvertType(input, input_logical_type); in ExpandInstruction()
89 input = BitcastConvertType(input, input_logical_type); in ExpandInstruction()
104 BitcastConvertType(input, to_shape.element_type()); in ExpandInstruction()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | topk_ops.cc | 50 xla::XlaOp input_sm32 = xla::BitcastConvertType(input, xla::S32); in CreateKthOrderStatisticComputation()
93 return xla::BitcastConvertType(top_r1, xla::F32); in CreateKthOrderStatisticComputation()
194 xla::XlaOp topk_r2_s32 = xla::BitcastConvertType(topk_r2, xla::S32); in CreateTopKUnique()
294 xla::XlaOp input_u32_r2 = xla::BitcastConvertType(input, xla::U32); in CreateMakeUnique()
305 return xla::BitcastConvertType(or_r2, xla::F32); in CreateMakeUnique()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 88 XlaOp in_s32 = BitcastConvertType(in, S32); in TopK()
102 BitcastConvertType(ConvertElementType(input, F32), S32)), in TopK()
122 BitcastConvertType( in TopK()
|
| D | prng.cc | 51 key[0] = BitcastConvertType(key[0], U32); in ThreeFry2x32()
52 key[1] = BitcastConvertType(key[1], U32); in ThreeFry2x32()
504 XlaOp range = BitcastConvertType(maxval, unsigned_type) - in ConvertRandomBitsToUniformInt()
505 BitcastConvertType(minval, unsigned_type); in ConvertRandomBitsToUniformInt()
510 return minval + BitcastConvertType(dist_div_2, type) + in ConvertRandomBitsToUniformInt()
511 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()
1361 auto from_as_int = BitcastConvertType(from, int_type); in NextAfter()
1362 auto to_as_int = BitcastConvertType(to, int_type); in NextAfter()
1371 result_for_nan = BitcastConvertType(result_for_nan, int_type); in NextAfter()
1429 return BitcastConvertType(result, shape.element_type()); in NextAfter()
|
| /external/tensorflow/tensorflow/compiler/xla/python/xla_extension/ |
| D | ops.pyi | 121 def BitcastConvertType(operand: XlaOp,
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 113 ops.def("BitcastConvertType", &BitcastConvertType, py::arg("operand"), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 839 ops.BitcastConvertType(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 815 XlaOp BitcastConvertType(XlaOp operand, PrimitiveType new_element_type);
1437 friend XlaOp BitcastConvertType(XlaOp operand,
2507 XlaOp BitcastConvertType(XlaOp operand, PrimitiveType new_element_type);
|
| D | xla_builder.cc | 2215 XlaOp XlaBuilder::BitcastConvertType(XlaOp operand, in BitcastConvertType() function in xla::XlaBuilder
4825 XlaOp BitcastConvertType(const XlaOp operand, PrimitiveType new_element_type) { in BitcastConvertType() function
4826 return operand.builder()->BitcastConvertType(operand, new_element_type); in BitcastConvertType()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_patterns.td | 706 // TODO(mgester): Due to restrictions of xla::BitcastConvertType we currently
|
| /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…
354 <b> `BitcastConvertType(operand, new_element_type)` </b>
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 812 value_map[op] = xla::BitcastConvertType( in ExportXlaOp()
|