Home
last modified time | relevance | path

Searched defs:rhs_type (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dbroadcast_utils.cc34 RankedTensorType rhs_type = rhs.getType().dyn_cast<RankedTensorType>(); in IsLegalNumpyRankedBroadcast() local
55 auto rhs_type = rhs.getType().dyn_cast<RankedTensorType>(); in ComputeBinaryElementwiseBroadcastingResultExtents() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_standard.cc43 auto rhs_type = rhs.getType().cast<TensorType>(); in matchAndRewrite() local
80 auto rhs_type = rhs.getType().cast<TensorType>(); in matchAndRewrite() local
Dlhlo_legalize_to_affine.cc51 MemRefType rhs_type = rhs.getType().cast<MemRefType>(); in matchAndRewrite() local
110 const auto& rhs_type = rhs.getType().template cast<MemRefType>(); in matchAndRewrite() local
Dchlo_legalize_to_hlo.cc1009 auto rhs_type = in matchAndRewrite() local
1059 auto rhs_type = rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
Dtransform_unranked_hlo.cc226 auto rhs_type = rhs.getType().dyn_cast<UnrankedTensorType>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/
Doperand_upcaster_test.cc39 PrimitiveType lhs_type, rhs_type, result_type; in TEST_P() local
Dshape_inference_test.cc621 ConvolveArgs MakeConvolveArgs(PrimitiveType lhs_type, PrimitiveType rhs_type) { in MakeConvolveArgs()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbatchmatmul_to_einsum.cc57 auto rhs_type = input_rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
Dunroll_batch_matmul.cc205 auto rhs_type = input_rhs.getType().cast<RankedTensorType>(); in matchAndRewrite() local
Deinsum.cc242 RankedTensorType rhs_type = rhs->getType().cast<RankedTensorType>(); in reshapeForBatchMatmul() local
Dlegalize_hlo.cc453 auto rhs_type = rhs.getType().cast<ShapedType>(); in ConvertDot() local
/external/arm-trusted-firmware/plat/common/
Dubsan.c40 struct type_descriptor *rhs_type; member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc208 auto rhs_type = DropRefType(std::get<1>(types)); in BroadcastCompatible() local
Dtf_ops_a_m.cc1250 auto rhs_type = RankedTensorType::get(rhs_dims, ranked.getElementType()); in GetHoistParams() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dchlo_ops.cc137 ShapedType rhs_type = operands[1].getType().dyn_cast<ShapedType>(); in InferBroadcastBinaryOpReturnTypeComponents() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc971 llvm::Type* rhs_type = rhs->getType(); in GetGemvBuffersWithCanonicalType() local
/external/mesa3d/src/compiler/glsl/
Dast_function.cpp1528 const glsl_type *rhs_type = in emit_inline_vector_constructor() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc592 RankedTensorType rhs_type = rhs_val.getType().cast<RankedTensorType>(); in convertConcatV2Op() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2783 auto rhs_type = rhs.getType().cast<RankedTensorType>(); in BroadcastBatchMatMulV2Operands() local
2835 auto rhs_type = rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/external/libabigail/src/
Dabg-ir.cc20100 const type_base& rhs_type) in types_are_being_compared()
20153 const type_base& rhs_type) in maybe_propagate_canonical_type()