Home
last modified time | relevance | path

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

12

/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/
Dbatch_norm_op.cc59 input = xla::ConvertElementType(input, scale_type); in Compile()
67 ctx->SetOutput(0, xla::ConvertElementType(xla::GetTupleElement(output, 0), in Compile()
93 ctx->SetOutput(0, xla::ConvertElementType(output, input_type)); in Compile()
134 XlaHelpers::ConvertElementType(ctx->Input(0), scale_dtype); in Compile()
136 XlaHelpers::ConvertElementType(ctx->Input(1), scale_dtype); in Compile()
186 XlaHelpers::ConvertElementType(grad_backprop, accumulation_type); in Compile()
190 offset_backprop = XlaHelpers::ConvertElementType(reduce, scale_dtype); in Compile()
199 converted = XlaHelpers::ConvertElementType(mul, accumulation_type); in Compile()
203 auto scratch2 = XlaHelpers::ConvertElementType(reduce, scale_dtype); in Compile()
210 ctx->SetOutput(0, XlaHelpers::ConvertElementType(x_backprop, input_dtype)); in Compile()
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()
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()
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.cc70 xla::ConvertElementType(exp_shifted, xla_accumulation_type); in Compile()
74 auto sum = XlaHelpers::ConvertElementType(reduce, type); in Compile()
114 XlaHelpers::ConvertElementType(exp_shifted_logits, accumulation_type); in CrossEntropyWithLogits()
118 auto sum_exp = XlaHelpers::ConvertElementType(reduce, type); in CrossEntropyWithLogits()
129 auto sum = xla::Reduce(XlaHelpers::ConvertElementType(mul, accumulation_type), in CrossEntropyWithLogits()
132 auto loss = XlaHelpers::ConvertElementType(sum, type); in CrossEntropyWithLogits()
Dreduction_ops_common.cc44 return XlaHelpers::ConvertElementType(reduce_output, input_type(0)); in BuildFinalizer()
109 auto data = xla::ConvertElementType(ctx->Input(0), type); in Compile()
111 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()
Dquantize_and_dequantize_op.cc97 -Pow(two, ConvertElementType(num_bits - xla::ConstantR0<int32>(b, 1), in Compile()
100 Pow(two, ConvertElementType(num_bits - xla::ConstantR0<int32>(b, 1), in Compile()
105 max_quantized = Pow(two, ConvertElementType(num_bits, xla_type)) - one; in Compile()
Dimage_ops.cc193 xla::XlaOp factor = XlaHelpers::ConvertElementType(context->Input(1), type); in Compile()
196 auto converted = XlaHelpers::ConvertElementType(input, accumulation_type); in Compile()
203 output = XlaHelpers::ConvertElementType(output, type); in Compile()
240 XlaHelpers::ConvertElementType(context->Input(0), DT_FLOAT); in Compile()
242 XlaHelpers::ConvertElementType(context->Input(1), DT_FLOAT); in Compile()
265 auto output = XlaHelpers::ConvertElementType( in Compile()
294 XlaHelpers::ConvertElementType(context->Input(0), DT_FLOAT); in Compile()
296 XlaHelpers::ConvertElementType(context->Input(1), DT_FLOAT); in Compile()
324 auto output = XlaHelpers::ConvertElementType( in Compile()
Dmatmul_op.cc78 a = xla::ConvertElementType(a, xla::F32); in Compile()
81 b = xla::ConvertElementType(b, xla::F32); 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.cc105 XlaHelpers::ConvertElementType(ctx->Input(0), dtype), init, *reducer, in Compile()
108 output = XlaHelpers::ConvertElementType(output, ctx->input_type(0)); in Compile()
Dfake_quantize_ops.cc263 XlaHelpers::ConvertElementType(select1, accumulation_type), in Compile()
266 xla::XlaOp output1 = XlaHelpers::ConvertElementType(reduce1, data_type); in Compile()
272 XlaHelpers::ConvertElementType(select2, accumulation_type), in Compile()
275 xla::XlaOp output2 = XlaHelpers::ConvertElementType(reduce2, data_type); in Compile()
Dpooling_ops.cc236 ConvertElementType(ctx->Input(0), xla_reduction_type_); in Compile()
241 ctx->SetOutput(0, ConvertElementType(pooling, ctx->input_xla_type(0))); in Compile()
428 xla::ConvertElementType(out_backprop, xla_reduction_type); in Compile()
443 xla::ConvertElementType(in_backprop, xla_out_backprop_type)); in Compile()
563 xla::ConvertElementType(xla::ConvertElementType(input, xla::BF16), in Compile()
Dcast_op.cc52 output = xla::ConvertElementType(xla::Real(input), dst_type_); in Compile()
84 output = xla::ConvertElementType(input, dst_type_); in Compile()
Dreduction_ops.cc128 divisor = xla::ConvertElementType(divisor, xla_reduction_type_); in BuildFinalizer()
129 return XlaHelpers::ConvertElementType(reduce_output / divisor, in BuildFinalizer()
Dimage_resize_ops.cc146 return xla::ConvertElementType(xla::ConstantR1<float>(builder, kernel), type); in MakeBilinear1DKernel()
163 return xla::ConvertElementType(xla::ConstantR1<float>(builder, kernel), type); in MakeNearestNeighbor1DKernel()
498 input = xla::ConvertElementType(input, xla::F32); in GeneralCompile()
673 output = xla::ConvertElementType(output, output_type_); in Compile()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc132 auto fst = ConvertElementType(u64, U32); in Uint64ToUint32s()
133 auto snd = ConvertElementType(ShiftRightLogical(u64, const32), U32); in Uint64ToUint32s()
139 return ConvertElementType(u32s[0], U64) | in Uint32sToUint64()
140 ShiftLeft(ConvertElementType(u32s[1], U64), in Uint32sToUint64()
Dcomparators.cc131 first_lhs_param = ConvertElementType(first_lhs_param, F32); in CreateScalarComparisonComputation()
132 first_rhs_param = ConvertElementType(first_rhs_param, F32); in CreateScalarComparisonComputation()
Dqr.cc93 XlaOp x_after_k = Mul(x, ConvertElementType(Gt(iota, k), type), in House()
110 auto e_k = Broadcast(ConvertElementType(Eq(iota, k), type), in House()
204 auto predecessor_mask = ConvertElementType(Lt(iota, j), type); in QRBlock()
205 auto mask = Broadcast(ConvertElementType(Eq(iota, j), type), in QRBlock()
Dmath.cc43 operand = ConvertElementType(operand, F32); in DoWithUpcastToF32()
47 result = ConvertElementType(result, elem_ty); in DoWithUpcastToF32()
125 return Eq(BitcastConvertType(ConvertElementType(operand, F32), U32), in IsNegZero()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.cc117 xla::XlaOp XlaHelpers::ConvertElementType(const xla::XlaOp& operand, in ConvertElementType() function in tensorflow::XlaHelpers
121 return xla::ConvertElementType(operand, convert_to); in ConvertElementType()
Dxla_helpers.h73 static xla::XlaOp ConvertElementType(const xla::XlaOp& operand,

12