/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
D | broadcast_utils.cc | 33 RankedTensorType lhs_type = lhs.getType().dyn_cast<RankedTensorType>(); in IsLegalNumpyRankedBroadcast() local 35 if (!lhs_type || !rhs_type) return false; in IsLegalNumpyRankedBroadcast() 36 if (lhs_type.getRank() == rhs_type.getRank()) return true; in IsLegalNumpyRankedBroadcast() 39 auto smaller_rank = std::min(lhs_type.getRank(), rhs_type.getRank()); in IsLegalNumpyRankedBroadcast() 40 auto larger_rank = std::max(lhs_type.getRank(), rhs_type.getRank()); in IsLegalNumpyRankedBroadcast() 54 auto lhs_type = lhs.getType().dyn_cast<RankedTensorType>(); in ComputeBinaryElementwiseBroadcastingResultExtents() local 56 if (!lhs_type || !rhs_type) { in ComputeBinaryElementwiseBroadcastingResultExtents() 66 int64_t result_rank = std::max(lhs_type.getRank(), rhs_type.getRank()); in ComputeBinaryElementwiseBroadcastingResultExtents()
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
D | lhlo_legalize_to_affine.cc | 50 MemRefType lhs_type = lhs.getType().cast<MemRefType>(); in matchAndRewrite() local 52 Type element_type = lhs_type.getElementType(); in matchAndRewrite() 53 ArrayRef<int64_t> shape_lhs = lhs_type.getShape(); in matchAndRewrite() 56 if ((lhs_type.getRank() != 2) || (rhs_type.getRank() != 2)) { in matchAndRewrite() 109 const auto& lhs_type = lhs.getType().template cast<MemRefType>(); in matchAndRewrite() local 111 const auto& element_type = lhs_type.getElementType(); in matchAndRewrite() 113 if (lhs_type.getShape() != rhs_type.getShape()) { in matchAndRewrite() 129 BuildBoundedAffineLoopNest(rewriter, op.getLoc(), lhs_type.getShape(), in matchAndRewrite()
|
D | legalize_to_standard.cc | 42 auto lhs_type = lhs.getType().cast<TensorType>(); in matchAndRewrite() local 46 if (lhs_type.getShape() != rhs_type.getShape()) return failure(); in matchAndRewrite() 48 if (!lhs_type.getElementType().isSignlessInteger() || in matchAndRewrite() 79 auto lhs_type = lhs.getType().cast<TensorType>(); in matchAndRewrite() local 83 if (lhs_type.getShape() != rhs_type.getShape()) return failure(); in matchAndRewrite() 85 if (!lhs_type.getElementType().isa<FloatType>() || in matchAndRewrite()
|
D | chlo_legalize_to_hlo.cc | 1007 auto lhs_type = in matchAndRewrite() local 1011 if (!lhs_type || !rhs_type) return failure(); in matchAndRewrite() 1014 if (lhs_type.getRank() != rhs_type.getRank()) return failure(); in matchAndRewrite() 1017 if (!lhs_type.hasStaticShape() || !rhs_type.hasStaticShape()) in matchAndRewrite() 1020 for (auto extents : llvm::zip(lhs_type.getShape(), rhs_type.getShape())) { in matchAndRewrite() 1058 auto lhs_type = lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local 1062 if (!lhs_type || !rhs_type || !result_type) return failure(); in matchAndRewrite() 1095 int64_t result_rank = std::max(lhs_type.getRank(), rhs_type.getRank()); in matchAndRewrite() 1106 llvm::seq<int64_t>(result_rank - lhs_type.getRank(), result_rank)); in matchAndRewrite() 1110 lhs_type.getElementType()), in matchAndRewrite()
|
D | transform_unranked_hlo.cc | 225 auto lhs_type = lhs.getType().dyn_cast<UnrankedTensorType>(); in matchAndRewrite() local 231 if (!lhs_type || !rhs_type) return failure(); in matchAndRewrite() 243 loc, RankedTensorType::get({}, lhs_type.getElementType()), lhs); in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | operand_upcaster_test.cc | 39 PrimitiveType lhs_type, rhs_type, result_type; in TEST_P() local 40 std::tie(lhs_type, rhs_type, result_type) = GetParam(); in TEST_P() 51 module_tmpl, primitive_util::LowercasePrimitiveTypeName(lhs_type), in TEST_P() 57 EXPECT_EQ(upcasted, ShouldUpcast(lhs_type, result_type) || in TEST_P() 62 ShouldUpcast(lhs_type, result_type) in TEST_P()
|
D | shape_inference_test.cc | 621 ConvolveArgs MakeConvolveArgs(PrimitiveType lhs_type, PrimitiveType rhs_type) { in MakeConvolveArgs() argument 626 args.lhs_shape = ShapeUtil::MakeShape(lhs_type, {10, 11, 3, 4}); in MakeConvolveArgs()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | batchmatmul_to_einsum.cc | 56 auto lhs_type = input_lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local 59 if (!lhs_type || !rhs_type) return failure(); in matchAndRewrite() 61 auto lhs_shape = lhs_type.getShape(); in matchAndRewrite()
|
D | unroll_batch_matmul.cc | 204 auto lhs_type = input_lhs.getType().cast<RankedTensorType>(); in matchAndRewrite() local 207 auto element_type = lhs_type.getElementType(); in matchAndRewrite() 214 auto lhs_shape = lhs_type.getShape(); in matchAndRewrite() 231 lhs_type = input_lhs.getType().cast<RankedTensorType>(); in matchAndRewrite() 232 lhs_shape = lhs_type.getShape(); in matchAndRewrite()
|
D | einsum.cc | 241 RankedTensorType lhs_type = lhs->getType().cast<RankedTensorType>(); in reshapeForBatchMatmul() local 249 int64_t b = lhs_type.getShape()[std::get<0>(i)]; in reshapeForBatchMatmul() 261 int64_t b = lhs_type.getShape()[std::get<0>(i)]; in reshapeForBatchMatmul() 268 lhs_out_size *= lhs_type.getShape()[std::get<0>(i)]; in reshapeForBatchMatmul() 276 lhs_rhs_size *= lhs_type.getShape()[std::get<0>(i)]; in reshapeForBatchMatmul() 292 *lhs = createReshapeOp(*lhs, lhs_shape, lhs_type.getElementType(), loc, in reshapeForBatchMatmul()
|
D | legalize_hlo.cc | 452 auto lhs_type = lhs.getType().cast<ShapedType>(); in ConvertDot() local 454 const int lhs_rank = lhs_type.getRank(); in ConvertDot() 459 lhs_type, dot_dimension_numbers.lhs_batching_dimensions(), in ConvertDot() 477 RankedTensorType::get(lhs_transposed_shape, lhs_type.getElementType()), in ConvertDot() 510 RankedTensorType::get(lhs_flattened_shape, lhs_type.getElementType()), in ConvertDot()
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
D | chlo_ops.cc | 136 ShapedType lhs_type = operands[0].getType().dyn_cast<ShapedType>(); in InferBroadcastBinaryOpReturnTypeComponents() local 138 if (!lhs_type || !rhs_type || in InferBroadcastBinaryOpReturnTypeComponents() 139 lhs_type.getElementType() != rhs_type.getElementType()) { in InferBroadcastBinaryOpReturnTypeComponents() 142 if (!element_type) element_type = lhs_type.getElementType(); in InferBroadcastBinaryOpReturnTypeComponents() 144 GetBroadcastType(lhs_type, rhs_type, element_type, broadcast_dimensions); in InferBroadcastBinaryOpReturnTypeComponents() 198 ShapedType lhs_type = operands[0].getType().dyn_cast<ShapedType>(); in inferReturnTypeComponents() local 199 if (!lhs_type) { in inferReturnTypeComponents() 202 Type element_type = ComplexType::get(lhs_type.getElementType()); in inferReturnTypeComponents()
|
/external/deqp-deps/SPIRV-Tools/source/fuzz/ |
D | fuzzer_pass_add_equation_instructions.cpp | 219 auto lhs_type = in Apply() local 223 if (lhs_type->AsVector()) { in Apply() 224 lhs_element_count = lhs_type->AsVector()->element_count(); in Apply() 225 lhs_bit_width = lhs_type->AsVector() in Apply() 231 lhs_bit_width = lhs_type->AsInteger()->width(); in Apply()
|
/external/swiftshader/third_party/SPIRV-Tools/source/fuzz/ |
D | fuzzer_pass_add_equation_instructions.cpp | 219 auto lhs_type = in Apply() local 223 if (lhs_type->AsVector()) { in Apply() 224 lhs_element_count = lhs_type->AsVector()->element_count(); in Apply() 225 lhs_bit_width = lhs_type->AsVector() in Apply() 231 lhs_bit_width = lhs_type->AsInteger()->width(); in Apply()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
D | tf_types.cc | 207 auto lhs_type = DropRefType(std::get<0>(types)); in BroadcastCompatible() local 211 auto lhs_tt = lhs_type.dyn_cast<TensorType>(); in BroadcastCompatible() 214 if (lhs_type != rhs_type) return false; in BroadcastCompatible() 241 auto lhs_rt = lhs_type.dyn_cast<RankedTensorType>(); in BroadcastCompatible()
|
D | tf_ops_helpers.inc | 456 auto lhs_type = arithmetic_op.x().getType().template cast<ShapedType>(); 494 Type element_ty = lhs_type.getElementType(); 505 if (rhs_attr && is_valid_broadcasting(lhs_type, rhs_type, result_type)) { 512 is_valid_broadcasting(rhs_type, lhs_type, result_type)) {
|
D | tf_ops_a_m.cc | 1260 auto lhs_type = RankedTensorType::get(lhs_dims, ranked.getElementType()); in GetHoistParams() local 1265 lhs_type, in GetHoistParams()
|
/external/arm-trusted-firmware/plat/common/ |
D | ubsan.c | 39 struct type_descriptor *lhs_type; member
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | tiled_dot_emitter.cc | 970 llvm::Type* lhs_type = lhs->getType(); in GetGemvBuffersWithCanonicalType() local 976 b->CreateBitCast(lhs, GetPointerToElementType(lhs_type)); in GetGemvBuffersWithCanonicalType()
|
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
D | legalize_common.cc | 591 RankedTensorType lhs_type = lhs_val.getType().cast<RankedTensorType>(); in convertConcatV2Op() local 593 ArrayRef<int64_t> lhs_tensor_shape = lhs_type.getShape(); in convertConcatV2Op() 638 lhs_type.getElementType() in convertConcatV2Op() 667 RankedTensorType::get(lhs_type.getShape(), result_quant_type); in convertConcatV2Op()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
D | legalize_tf.cc | 2782 auto lhs_type = lhs.getType().cast<RankedTensorType>(); in BroadcastBatchMatMulV2Operands() local 2787 OpTrait::util::getBroadcastedShape(lhs_type.getShape().drop_back(2), in BroadcastBatchMatMulV2Operands() 2813 broadcast_one_side(lhs, lhs_type, lhs_splitted.tail(), out_lhs); in BroadcastBatchMatMulV2Operands() 2834 auto lhs_type = lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local 2836 if (!lhs_type || !rhs_type) return failure(); in matchAndRewrite() 2837 if (lhs_type.getElementType().isa<ComplexType>() && op.adj_x()) { in matchAndRewrite() 2838 lhs = rewriter.create<TF::ConjOp>(op.getLoc(), lhs_type, lhs); in matchAndRewrite() 2847 lhs_type = lhs.getType().cast<RankedTensorType>(); in matchAndRewrite() 2849 assert(lhs_type.getRank() == rhs_type.getRank()); in matchAndRewrite() 2850 int64_t rank = lhs_type.getRank(); in matchAndRewrite()
|
/external/libabigail/src/ |
D | abg-ir.cc | 12432 maybe_propagate_canonical_type(const type_base& lhs_type, 20099 types_are_being_compared(const type_base& lhs_type, in types_are_being_compared() argument 20102 type_base *l = &const_cast<type_base&>(lhs_type); in types_are_being_compared() 20152 maybe_propagate_canonical_type(const type_base& lhs_type, in maybe_propagate_canonical_type() argument 20156 if (const environment *env = lhs_type.get_environment()) in maybe_propagate_canonical_type() 20158 if (type_base_sptr canonical_type = lhs_type.get_canonical_type()) in maybe_propagate_canonical_type() 20160 && !types_are_being_compared(lhs_type, rhs_type)) in maybe_propagate_canonical_type()
|