Searched refs:is_float16 (Results 1 – 14 of 14) sorted by relevance
67 is_float16 = tmp_cpu.dtype == np.float1668 if is_float16:72 if is_float16:
698 is_float16 = param.data.dtype == numpy.float16699 if is_float16:703 if is_float16:754 is_float16 = recvbuf.dtype == numpy.float16757 elif is_float16:764 if is_float16:772 if is_float16:
311 CHECK(i >= 0 && i < (t.is_float16() ? 8 : 4)); in PrintVecElemLoad()324 } else if (t.is_float16()) { in PrintVecElemLoad()335 CHECK(i >= 0 && i < (t.is_float16() ? 8 : 4)); in PrintVecElemStore()348 } else if (t.is_float16()) { in PrintVecElemStore()653 if (op->dtype.is_float16()) { in VisitExpr_()837 if (op->dtype.is_float16() && IsVolatile(op->buffer_var.get())) { in HandleVolatileLoads()859 if (t.is_float16()) { in PrintVecElemLoadExpr()
129 is_float16
139 is_float16
140 def is_float16(t): function
142 def is_float16(t): function
200 def is_float16(t): function
95 bool is_float16() const { return is_float() && bits() == 16; } in is_float16() function
501 if (ty.is_float16()) return ty.lanes() > 2; in is_warp_reduction()
523 (pa.float16(), types.is_float16),
695 (pa.float16(), types.is_float16),
736 (pa.float16(), types.is_float16),