Home
last modified time | relevance | path

Searched refs:gte (Results 1 – 25 of 90) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter.cc81 HloInstruction *gte = nullptr, *add1 = nullptr, *add2 = nullptr, in FindConvWithRelu() local
88 &gte, Op(&conv_instr).WithOpcode(HloOpcode::kCustomCall), 0); in FindConvWithRelu()
143 {conv_instr, bias_broadcast_instr, gte, add1, add2, mul1, mul2}) { in FindConvWithRelu()
300 HloInstruction* gte; member
315 HloInstruction* gte = nullptr; in FindConvWithClampAndConvertToInt8() local
322 &gte, Op(&conv_instr).WithOpcode(HloOpcode::kCustomCall), 0), in FindConvWithClampAndConvertToInt8()
330 return ConvWithConvertOrClamp{instr, gte, conv}; in FindConvWithClampAndConvertToInt8()
341 auto gte = match.gte; in RewriteForConvertOrClampImpl() local
348 gte->mutable_shape()->set_element_type(convert_out_type); in RewriteForConvertOrClampImpl()
352 TF_RETURN_IF_ERROR(convert_or_clamp->ReplaceAllUsesWithDifferentShape(gte)); in RewriteForConvertOrClampImpl()
[all …]
Dhlo_to_ir_bindings.cc101 llvm::Value* HloToIrBindings::EmitGetTupleElement(const HloInstruction* gte, in EmitGetTupleElement() argument
104 if (gte->operand(0)->opcode() != HloOpcode::kGetTupleElement) { in EmitGetTupleElement()
106 gte->shape(), gte->tuple_index(), /*alignment=*/1, in EmitGetTupleElement()
107 GetTypedIrValue(*gte->operand(0), {}, base_ptr), b_); in EmitGetTupleElement()
110 gte->shape(), gte->tuple_index(), /*alignment=*/1, in EmitGetTupleElement()
111 EmitGetTupleElement(gte->operand(0), base_ptr), b_); in EmitGetTupleElement()
/external/bouncycastle/repackaged/bcprov/src/main/java/com/android/org/bouncycastle/math/ec/custom/sec/
DSecP192K1Field.java30 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in add()
39 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in addExt()
51 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in addOne()
60 if (z[5] == P5 && Nat192.gte(z, P)) in fromBigInteger()
106 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in multiplyAddToExt()
154 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in reduce()
163 || (z[5] == P5 && Nat192.gte(z, P))) in reduce32()
215 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in twice()
DSecP256K1Field.java32 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in add()
41 if (c != 0 || (zz[15] == PExt15 && Nat.gte(16, zz, PExt))) in addExt()
53 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in addOne()
62 if (z[7] == P7 && Nat256.gte(z, P)) in fromBigInteger()
108 if (c != 0 || (zz[15] == PExt15 && Nat.gte(16, zz, PExt))) in multiplyAddToExt()
156 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in reduce()
165 || (z[7] == P7 && Nat256.gte(z, P))) in reduce32()
217 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in twice()
DSecP224K1Field.java31 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in add()
40 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in addExt()
52 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in addOne()
61 if (z[6] == P6 && Nat224.gte(z, P)) in fromBigInteger()
107 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in multiplyAddToExt()
155 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in reduce()
164 || (z[6] == P6 && Nat224.gte(z, P))) in reduce32()
216 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in twice()
DSecP224R1Field.java32 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in add()
41 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in addExt()
53 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in addOne()
62 if (z[6] == P6 && Nat224.gte(z, P)) in fromBigInteger()
108 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in multiplyAddToExt()
205 || (z[6] == P6 && Nat224.gte(z, P))) in reduce()
239 || (z[6] == P6 && Nat224.gte(z, P))) in reduce32()
291 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in twice()
DSecP192R1Field.java31 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in add()
40 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in addExt()
52 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in addOne()
61 if (z[5] == P5 && Nat192.gte(z, P)) in fromBigInteger()
107 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in multiplyAddToExt()
201 || (z[5] == P5 && Nat192.gte(z, P))) in reduce()
232 || (z[5] == P5 && Nat192.gte(z, P))) in reduce32()
284 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in twice()
DSecP256R1Field.java31 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in add()
40 if (c != 0 || ((zz[15] >>> 1) >= PExt15s1 && Nat.gte(16, zz, PExt))) in addExt()
49 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in addOne()
58 if (z[7] == P7 && Nat256.gte(z, P)) in fromBigInteger()
104 if (c != 0 || ((zz[15] >>> 1) >= PExt15s1 && Nat.gte(16, zz, PExt))) in multiplyAddToExt()
234 if (cc != 0 || (z[7] == P7 && Nat256.gte(z, P))) in reduce32()
283 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in twice()
DSecP384R1Field.java34 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in add()
43 if (c != 0 || (zz[23] == PExt23 && Nat.gte(24, zz, PExt))) in addExt()
55 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in addOne()
64 if (z[11] == P11 && Nat.gte(12, z, P)) in fromBigInteger()
230 || (z[11] == P11 && Nat.gte(12, z, P))) in reduce32()
282 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in twice()
/external/bouncycastle/bcprov/src/main/java/org/bouncycastle/math/ec/custom/sec/
DSecP224K1Field.java27 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in add()
36 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in addExt()
48 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in addOne()
57 if (z[6] == P6 && Nat224.gte(z, P)) in fromBigInteger()
103 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in multiplyAddToExt()
151 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in reduce()
160 || (z[6] == P6 && Nat224.gte(z, P))) in reduce32()
212 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in twice()
DSecP256K1Field.java28 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in add()
37 if (c != 0 || (zz[15] == PExt15 && Nat.gte(16, zz, PExt))) in addExt()
49 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in addOne()
58 if (z[7] == P7 && Nat256.gte(z, P)) in fromBigInteger()
104 if (c != 0 || (zz[15] == PExt15 && Nat.gte(16, zz, PExt))) in multiplyAddToExt()
152 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in reduce()
161 || (z[7] == P7 && Nat256.gte(z, P))) in reduce32()
213 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in twice()
DSecP192K1Field.java26 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in add()
35 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in addExt()
47 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in addOne()
56 if (z[5] == P5 && Nat192.gte(z, P)) in fromBigInteger()
102 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in multiplyAddToExt()
150 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in reduce()
159 || (z[5] == P5 && Nat192.gte(z, P))) in reduce32()
211 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in twice()
DSecP192R1Field.java27 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in add()
36 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in addExt()
48 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in addOne()
57 if (z[5] == P5 && Nat192.gte(z, P)) in fromBigInteger()
103 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in multiplyAddToExt()
197 || (z[5] == P5 && Nat192.gte(z, P))) in reduce()
228 || (z[5] == P5 && Nat192.gte(z, P))) in reduce32()
280 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in twice()
DSecP224R1Field.java28 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in add()
37 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in addExt()
49 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in addOne()
58 if (z[6] == P6 && Nat224.gte(z, P)) in fromBigInteger()
104 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in multiplyAddToExt()
201 || (z[6] == P6 && Nat224.gte(z, P))) in reduce()
235 || (z[6] == P6 && Nat224.gte(z, P))) in reduce32()
287 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in twice()
DSecP256R1Field.java27 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in add()
36 if (c != 0 || ((zz[15] >>> 1) >= PExt15s1 && Nat.gte(16, zz, PExt))) in addExt()
45 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in addOne()
54 if (z[7] == P7 && Nat256.gte(z, P)) in fromBigInteger()
100 if (c != 0 || ((zz[15] >>> 1) >= PExt15s1 && Nat.gte(16, zz, PExt))) in multiplyAddToExt()
230 if (cc != 0 || (z[7] == P7 && Nat256.gte(z, P))) in reduce32()
279 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in twice()
DSecP384R1Field.java30 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in add()
39 if (c != 0 || (zz[23] == PExt23 && Nat.gte(24, zz, PExt))) in addExt()
51 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in addOne()
60 if (z[11] == P11 && Nat.gte(12, z, P)) in fromBigInteger()
226 || (z[11] == P11 && Nat.gte(12, z, P))) in reduce32()
278 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in twice()
/external/bouncycastle/repackaged_platform/bcprov/src/main/java/com/android/internal/org/bouncycastle/math/ec/custom/sec/
DSecP192K1Field.java30 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in add()
39 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in addExt()
51 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in addOne()
60 if (z[5] == P5 && Nat192.gte(z, P)) in fromBigInteger()
106 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in multiplyAddToExt()
154 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in reduce()
163 || (z[5] == P5 && Nat192.gte(z, P))) in reduce32()
215 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in twice()
DSecP256K1Field.java32 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in add()
41 if (c != 0 || (zz[15] == PExt15 && Nat.gte(16, zz, PExt))) in addExt()
53 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in addOne()
62 if (z[7] == P7 && Nat256.gte(z, P)) in fromBigInteger()
108 if (c != 0 || (zz[15] == PExt15 && Nat.gte(16, zz, PExt))) in multiplyAddToExt()
156 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in reduce()
165 || (z[7] == P7 && Nat256.gte(z, P))) in reduce32()
217 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in twice()
DSecP224K1Field.java31 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in add()
40 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in addExt()
52 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in addOne()
61 if (z[6] == P6 && Nat224.gte(z, P)) in fromBigInteger()
107 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in multiplyAddToExt()
155 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in reduce()
164 || (z[6] == P6 && Nat224.gte(z, P))) in reduce32()
216 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in twice()
DSecP224R1Field.java32 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in add()
41 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in addExt()
53 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in addOne()
62 if (z[6] == P6 && Nat224.gte(z, P)) in fromBigInteger()
108 if (c != 0 || (zz[13] == PExt13 && Nat.gte(14, zz, PExt))) in multiplyAddToExt()
205 || (z[6] == P6 && Nat224.gte(z, P))) in reduce()
239 || (z[6] == P6 && Nat224.gte(z, P))) in reduce32()
291 if (c != 0 || (z[6] == P6 && Nat224.gte(z, P))) in twice()
DSecP192R1Field.java31 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in add()
40 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in addExt()
52 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in addOne()
61 if (z[5] == P5 && Nat192.gte(z, P)) in fromBigInteger()
107 if (c != 0 || (zz[11] == PExt11 && Nat.gte(12, zz, PExt))) in multiplyAddToExt()
201 || (z[5] == P5 && Nat192.gte(z, P))) in reduce()
232 || (z[5] == P5 && Nat192.gte(z, P))) in reduce32()
284 if (c != 0 || (z[5] == P5 && Nat192.gte(z, P))) in twice()
DSecP256R1Field.java31 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in add()
40 if (c != 0 || ((zz[15] >>> 1) >= PExt15s1 && Nat.gte(16, zz, PExt))) in addExt()
49 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in addOne()
58 if (z[7] == P7 && Nat256.gte(z, P)) in fromBigInteger()
104 if (c != 0 || ((zz[15] >>> 1) >= PExt15s1 && Nat.gte(16, zz, PExt))) in multiplyAddToExt()
234 if (cc != 0 || (z[7] == P7 && Nat256.gte(z, P))) in reduce32()
283 if (c != 0 || (z[7] == P7 && Nat256.gte(z, P))) in twice()
DSecP384R1Field.java34 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in add()
43 if (c != 0 || (zz[23] == PExt23 && Nat.gte(24, zz, PExt))) in addExt()
55 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in addOne()
64 if (z[11] == P11 && Nat.gte(12, z, P)) in fromBigInteger()
230 || (z[11] == P11 && Nat.gte(12, z, P))) in reduce32()
282 if (c != 0 || (z[11] == P11 && Nat.gte(12, z, P))) in twice()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtopk_rewriter.cc97 for (HloInstruction* gte : sort->users()) { in Run()
98 if (gte->opcode() != HloOpcode::kGetTupleElement || in Run()
99 gte->user_count() != 1) { in Run()
103 const HloInstruction* slice = gte->users()[0]; in Run()
176 for (HloInstruction* gte : sort->users()) { in Run()
177 for (HloInstruction* slice : gte->users()) { in Run()
178 if (gte->tuple_index() == 0) { in Run()
180 } else if (gte->tuple_index() == 1) { in Run()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_control_flow_to_scf.cc82 if (auto gte = val.getDefiningOp<GetTupleElementOp>()) { in MatchAndRewrite() local
83 if (!gte.getOperand().isa<mlir::BlockArgument>()) return {nullptr, 0}; in MatchAndRewrite()
84 int index = gte.index(); in MatchAndRewrite()
150 auto gte = cast<GetTupleElementOp>(use); in MatchAndRewrite() local
153 if (gte.index() == loopIndVar.second) { in MatchAndRewrite()
158 int index = gte.index(); in MatchAndRewrite()

1234