| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 852 def _emit_binary_op(self, op, lhs, lhs_ty, rhs, rhs_ty): argument
853 assert lhs_ty, rhs_ty
865 if lhs_ty == TFRTypes.I64 or lhs_ty == TFRTypes.I32:
867 elif lhs_ty == TFRTypes.F32:
874 '\n{} = {}{} {}, {} : {}'.format(ret, code, suffix, lhs, rhs, lhs_ty),
876 return ret, lhs_ty
879 lhs, lhs_ty = self.visit(node.target)
881 ret, ret_ty = self._emit_binary_op(node.op, lhs, lhs_ty, rhs, rhs_ty)
885 lhs, lhs_ty = self.visit(node.left)
887 return self._emit_binary_op(node.op, lhs, lhs_ty, rhs, rhs_ty)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | einsum.cc | 292 int rhs_named_label_count, llvm::StringRef out, RankedTensorType lhs_ty, in FlattenEllipsis() argument
303 new_labels = GenerateLabels(lhs_ty.getRank() - lhs_named_label_count, in FlattenEllipsis()
346 llvm::StringRef equation, RankedTensorType lhs_ty, in GetEinsumDimensionNumbers() argument
365 FlattenEllipsis(lhs, lhs_named_label, rhs, rhs_named_label, out, lhs_ty, in GetEinsumDimensionNumbers()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 3308 RankedTensorType lhs_ty = lhs.getType().dyn_cast<RankedTensorType>(); in inferReturnTypeComponents() local
3310 if (!lhs_ty || !rhs_ty) return set_unranked_results(); in inferReturnTypeComponents()
3312 int64_t lhs_rank = lhs_ty.getRank(); in inferReturnTypeComponents()
3327 inferredReturnShapes.emplace_back(lhs_ty.cast<ShapedType>()); in inferReturnTypeComponents()
3333 RankedTensorType min_rank_ty = broadcast_lhs ? lhs_ty : rhs_ty; in inferReturnTypeComponents()
3334 RankedTensorType max_rank_ty = broadcast_lhs ? rhs_ty : lhs_ty; in inferReturnTypeComponents()
3359 inferredReturnShapes.emplace_back(broadcast_shape, lhs_ty.getElementType()); in inferReturnTypeComponents()
3362 inferredReturnShapes.emplace_back(lhs_ty.cast<ShapedType>()); in inferReturnTypeComponents()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1359 RankedTensorType lhs_ty = lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
1364 if (!lhs_ty || !rhs_ty) return failure(); in matchAndRewrite()
1366 bool lhs_is_qtype = lhs_ty.getElementType().isa<mlir::quant::QuantizedType>(); in matchAndRewrite()
1377 auto batch_dims = lhs_ty.getShape().drop_back(2); in matchAndRewrite()
1386 auto lhs_shape_end = lhs_ty.getShape().take_back(2); in matchAndRewrite()
1394 UnrankedTensorType::get(lhs_ty.getElementType()), lhs, in matchAndRewrite()
1400 lhs_ty = lhs.getType().cast<RankedTensorType>(); in matchAndRewrite()
1408 Type output_type = UnrankedTensorType::get(lhs_ty.getElementType()); in matchAndRewrite()
1449 matmul = buildRescaleOpConvOutput(rewriter, op, matmul, lhs_ty, rhs_ty, in matchAndRewrite()
|
| /external/rust/crates/syn/src/ |
| D | generics.rs | 585 pub lhs_ty: Type,
1357 self.lhs_ty.to_tokens(tokens); in to_tokens()
|
| /external/rust/crates/syn/tests/common/ |
| D | eq.rs | 468 spanless_eq_struct!(WhereEqPredicate; span lhs_ty rhs_ty);
|
| /external/rust/crates/syn/src/gen/ |
| D | clone.rs | 1692 lhs_ty: self.lhs_ty.clone(), in clone()
|
| D | eq.rs | 1620 self.lhs_ty == other.lhs_ty && self.rhs_ty == other.rhs_ty in eq()
|
| D | fold.rs | 2722 lhs_ty: f.fold_type(node.lhs_ty), in fold_predicate_eq()
|
| D | hash.rs | 2158 self.lhs_ty.hash(state); in hash()
|
| D | debug.rs | 2312 formatter.field("lhs_ty", &self.lhs_ty); in fmt()
|
| D | visit_mut.rs | 3062 v.visit_type_mut(&mut node.lhs_ty); in visit_predicate_eq_mut()
|
| D | visit.rs | 3062 v.visit_type(&node.lhs_ty); in visit_predicate_eq()
|
| /external/rust/crates/syn/tests/debug/ |
| D | gen.rs | 4239 formatter.field("lhs_ty", Lite(&_val.lhs_ty)); in fmt()
|