Home
last modified time | relevance | path

Searched refs:ConvertElementType (Results 1 – 25 of 62) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconvert_test.cc51 ConvertElementType(a, S32); in TEST_F()
60 ConvertElementType(a, U32); in TEST_F()
69 ConvertElementType(a, PRED); in TEST_F()
78 ConvertElementType(a, U32); in TEST_F()
87 ConvertElementType(a, S32); in TEST_F()
96 ConvertElementType(a, PRED); in TEST_F()
105 ConvertElementType(a, F32); in TEST_F()
114 ConvertElementType(a, PRED); in TEST_F()
123 ConvertElementType(a, F32); in TEST_F()
132 ConvertElementType(a, S32); in TEST_F()
[all …]
Dunary_op_test.cc130 Sub(Mul(sign, ConvertElementType(abs, C64)), arg); in SignAbsTestHelper()
158 Add(Add(absc, absf0), Add(absf, ConvertElementType(absi, F32))); in XLA_TEST_F()
173 Add(sgnc, ConvertElementType( in XLA_TEST_F()
174 Add(Add(sgnf0, sgnf), ConvertElementType(sgni, F32)), C64)); in XLA_TEST_F()
207 ConvertElementType(Eq(lhs, rhs), S32); in XLA_TEST_F()
216 ConvertElementType(Eq(lhs, rhs), F32); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Drandom_ops_util.cc26 ConvertElementType(ConvertElementType(seed0, xla::U32), xla::U64); in GetU64FromS32Seeds()
28 ConvertElementType(ConvertElementType(seed1, xla::U32), xla::U64); in GetU64FromS32Seeds()
Dbucketize_op.cc45 input = xla::ConvertElementType(input, xla::F64); in Compile()
46 boundaries = xla::ConvertElementType(boundaries, xla::F64); in Compile()
48 input = xla::ConvertElementType(input, xla::F32); in Compile()
51 xla::ConvertElementType(xla::Ge(xla::Broadcast(input, {1}), boundaries, in Compile()
Dbatch_norm_op.cc91 input = xla::ConvertElementType(input, scale_type); in CompileImpl()
101 xla::ConvertElementType(xla::GetTupleElement(output, 0), input_type); in CompileImpl()
178 xla::XlaOp converted = xla::ConvertElementType(output, input_type); in CompileImpl()
260 XlaHelpers::ConvertElementType(ctx->Input(0), scale_dtype); in Compile()
262 XlaHelpers::ConvertElementType(ctx->Input(1), scale_dtype); in Compile()
312 XlaHelpers::ConvertElementType(grad_backprop, accumulation_type); in Compile()
316 offset_backprop = XlaHelpers::ConvertElementType(reduce, scale_dtype); in Compile()
325 converted = XlaHelpers::ConvertElementType(mul, accumulation_type); in Compile()
329 auto scratch2 = XlaHelpers::ConvertElementType(reduce, scale_dtype); in Compile()
336 ctx->SetOutput(0, XlaHelpers::ConvertElementType(x_backprop, input_dtype)); in Compile()
Dlrn_ops.cc52 auto converted = XlaHelpers::ConvertElementType(input, accumulation_type); in Compile()
59 auto sqr_sum = XlaHelpers::ConvertElementType(reduce, input_type(0)); in Compile()
139 XlaHelpers::ConvertElementType(in_image, accumulation_type); in Compile()
146 auto sqr_sum = XlaHelpers::ConvertElementType(reduce, input_type(0)); in Compile()
157 auto converted_dy = XlaHelpers::ConvertElementType(dy, accumulation_type); in Compile()
163 auto dy_reduced = XlaHelpers::ConvertElementType(dy_reduce, input_type(0)); in Compile()
Ddequantize_op.cc67 xla::XlaOp output = xla::ConvertElementType(input, xla::F32); in Compile()
68 xla::XlaOp min_range = xla::ConvertElementType(ctx->Input(1), xla::F32); in Compile()
69 xla::XlaOp max_range = xla::ConvertElementType(ctx->Input(2), xla::F32); in Compile()
89 output = xla::ConvertElementType(output, xla::BF16); in Compile()
Dwhere_op.cc50 xla::XlaOp zeros_int = xla::ConvertElementType(zeros, xla::S32); in Compile()
52 xla::ConvertElementType(reshaped_condition, xla::S32); in Compile()
53 xla::XlaOp compared = xla::ConvertElementType( in Compile()
81 result = xla::ConvertElementType(result, ctx->output_xla_type(0)); in Compile()
Dresampler_ops.cc57 first_term = xla::ConvertElementType(first_term, xla_type); in BilinearWeights()
89 sign_change = xla::ConvertElementType(sign_change, xla_type); in BilinearWeights()
202 xla::ConvertElementType( in BoundSamples()
207 xla::ConvertElementType( in BoundSamples()
412 xla::ConvertElementType( in CalculateGradWarp()
420 xla::ConvertElementType( in CalculateGradWarp()
428 xla::ConvertElementType( in CalculateGradWarp()
436 xla::ConvertElementType( in CalculateGradWarp()
517 XlaOp top_left = xla::ConvertElementType(warp, xla::S32); in Compile()
526 XlaOp ratio = warp - xla::ConvertElementType(top_left, data_type); in Compile()
[all …]
Dsoftmax_op.cc95 xla::ConvertElementType(exp_shifted, xla_accumulation_type); in Compile()
99 auto sum = XlaHelpers::ConvertElementType(reduce, type); in Compile()
139 XlaHelpers::ConvertElementType(exp_shifted_logits, accumulation_type); in CrossEntropyWithLogits()
143 auto sum_exp = XlaHelpers::ConvertElementType(reduce, type); in CrossEntropyWithLogits()
158 auto sum = xla::Reduce(XlaHelpers::ConvertElementType(mul, accumulation_type), in CrossEntropyWithLogits()
161 auto loss = XlaHelpers::ConvertElementType(sum, type); in CrossEntropyWithLogits()
Dreduction_ops_common.cc44 return XlaHelpers::ConvertElementType(reduce_output, input_type(0)); in BuildFinalizer()
114 auto data = xla::ConvertElementType(ctx->Input(0), type); in Compile()
116 auto initial = xla::ConvertElementType(InitialValue(b), type); in Compile()
Dl2loss_op.cc39 auto t = XlaHelpers::ConvertElementType(ctx->Input(0), accumulation_type); in Compile()
43 auto deconverted = XlaHelpers::ConvertElementType(reduce, dtype); in Compile()
Dcast_op.cc57 output = xla::ConvertElementType(xla::Real(input), dst_type_); in Compile()
89 output = xla::ConvertElementType(input, dst_type_); in Compile()
170 input = xla::ConvertElementType(input, output_logical_type); in Compile()
185 input = xla::ConvertElementType(input, dst_type_); in Compile()
Dquantize_and_dequantize_op.cc125 -Pow(two, ConvertElementType( in Compile()
130 -Pow(two, ConvertElementType( in Compile()
134 Pow(two, ConvertElementType(num_bits - xla::ConstantR0<int32>(b, 1), in Compile()
139 max_quantized = Pow(two, ConvertElementType(num_bits, xla_type)) - one; in Compile()
Dmatmul_op.cc81 a = xla::ConvertElementType(a, xla::F32); in Compile()
84 b = xla::ConvertElementType(b, xla::F32); in Compile()
Dimage_ops.cc195 xla::XlaOp factor = XlaHelpers::ConvertElementType(context->Input(1), type); in Compile()
198 auto converted = XlaHelpers::ConvertElementType(input, accumulation_type); in Compile()
205 output = XlaHelpers::ConvertElementType(output, type); in Compile()
242 XlaHelpers::ConvertElementType(context->Input(0), DT_FLOAT); in Compile()
244 XlaHelpers::ConvertElementType(context->Input(1), DT_FLOAT); in Compile()
267 auto output = XlaHelpers::ConvertElementType( in Compile()
296 XlaHelpers::ConvertElementType(context->Input(0), DT_FLOAT); in Compile()
298 XlaHelpers::ConvertElementType(context->Input(1), DT_FLOAT); in Compile()
326 auto output = XlaHelpers::ConvertElementType( in Compile()
Dbias_ops.cc114 XlaHelpers::ConvertElementType(ctx->Input(0), accumulation_type); in Compile()
118 ctx->SetOutput(0, XlaHelpers::ConvertElementType(reduce, input_type(0))); in Compile()
Dscan_ops.cc103 XlaHelpers::ConvertElementType(ctx->Input(0), dtype), init, *reducer, in Compile()
106 output = XlaHelpers::ConvertElementType(output, ctx->input_type(0)); in Compile()
Dshape_op.cc51 xla::ConvertElementType(xla::GetDimensionSize(ctx->Input(0), i), in Compile()
162 xla::ConvertElementType(xla::GetDimensionSize(ctx->Input(i), dim), in Compile()
218 size, xla::ConvertElementType(xla::GetDimensionSize(ctx->Input(0), i), in Compile()
221 size = xla::ConvertElementType(size, ctx->output_xla_type(0)); in Compile()
Dreplica_id_op.cc34 0, xla::ConvertElementType(xla::ReplicaId(ctx->builder()), xla::S32)); in Compile()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dlogdet.cc52 ConvertElementType(And(in_lower_triangle, is_zero), S32), "...a->..."); in LogDet()
57 Einsum(ConvertElementType(Eq(rows, num_zeros_in_triangle_per_row), S32), in LogDet()
72 Select(ConvertElementType(Rem(num_rows - num_triangle_rows, in LogDet()
Dprng.cc125 XlaOp fst = ConvertElementType(u64, U32); in Uint64ToUint32s()
126 XlaOp snd = ConvertElementType(ShiftRightLogical(u64, const32), U32); in Uint64ToUint32s()
133 return ConvertElementType(u32s[0], U64) | in Uint32sToUint64()
134 ShiftLeft(ConvertElementType(u32s[1], U64), in Uint32sToUint64()
285 ConvertElementType(x, U64) * ConstantR0<uint64>(x.builder(), k); in Philox4x32()
286 auto low = ConvertElementType(product, U32); in Philox4x32()
288 ConvertElementType(product >> ConstantR0<uint64>(x.builder(), 32), U32); in Philox4x32()
318 XlaOp key0 = ConvertElementType(key[0], U64); in ScramblePhiloxKey()
319 XlaOp key1 = ConvertElementType(key[1], U64); in ScramblePhiloxKey()
322 ConvertElementType(key0, U32), in ScramblePhiloxKey()
[all …]
Dmath.cc75 operand = ConvertElementType(operand, F32); in DoWithUpcastToF32()
79 result = ConvertElementType(result, elem_ty); in DoWithUpcastToF32()
157 return Eq(BitcastConvertType(ConvertElementType(operand, F32), U32), in IsNegZero()
961 a = ConvertElementType(a, F32); in Igamma()
962 x = ConvertElementType(x, F32); in Igamma()
967 result = ConvertElementType(result, a_shape.element_type()); in Igamma()
1008 a = ConvertElementType(a, F32); in IgammaGradA()
1009 x = ConvertElementType(x, F32); in IgammaGradA()
1013 result = ConvertElementType(result, a_shape.element_type()); in IgammaGradA()
1055 a = ConvertElementType(a, F32); in RandomGammaGrad()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dqr_expander.cc98 XlaOp x_after_k = Mul(x, ConvertElementType(Gt(iota, k), type), in House()
140 alpha - ConvertElementType(*beta, type)); in House()
142 auto e_k = Broadcast(ConvertElementType(Eq(iota, k), type), in House()
231 auto predecessor_mask = ConvertElementType(Lt(iota, j), type); in QrBlock()
232 auto mask = Broadcast(ConvertElementType(Eq(iota, j), type), in QrBlock()
237 Mul(ConvertElementType(beta, type), mask, in QrBlock()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dimage_resize_ops.cc60 0, xla::ConvertElementType(ctx->Input(0), ctx->output_xla_type(0))); in CompileGrad()
81 0, xla::ConvertElementType(ctx->Input(0), ctx->output_xla_type(0))); in CompileForward()

123