Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_standard.cc44 auto rhs_type = rhs.getType().cast<TensorType>(); in matchAndRewrite() local
81 auto rhs_type = rhs.getType().cast<TensorType>(); in matchAndRewrite() local
Dlhlo_legalize_to_affine.cc52 MemRefType rhs_type = rhs.getType().cast<MemRefType>(); in matchAndRewrite() local
489 const auto& rhs_type = rhs.getType().template cast<MemRefType>(); in matchAndRewrite() local
Dchlo_legalize_to_hlo.cc1321 auto rhs_type = in matchAndRewrite() local
1370 auto rhs_type = rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dresult_caster_test.cc34 PrimitiveType lhs_type, rhs_type, result_type; in TEST_P() local
Doperand_upcaster_test.cc39 PrimitiveType lhs_type, rhs_type, result_type; in TEST_P() local
Dshape_inference_test.cc645 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.cc219 auto rhs_type = input_rhs.getType().cast<RankedTensorType>(); in matchAndRewrite() local
Deinsum.cc398 RankedTensorType rhs_type = rhs->getType().cast<RankedTensorType>(); in reshapeForBatchMatmul() local
Dlegalize_hlo.cc750 auto rhs_type = rhs.getType().cast<ShapedType>(); in ConvertDot() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dbroadcast_utils.cc34 RankedTensorType rhs_type = rhs.getType().dyn_cast<RankedTensorType>(); in IsLegalNumpyRankedBroadcast() local
/external/arm-trusted-firmware/plat/common/
Dubsan.c40 struct type_descriptor *rhs_type; member
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dchlo_ops.cc136 ShapedType rhs_type = operands[1].getType().dyn_cast<ShapedType>(); in InferBroadcastBinaryOpReturnTypeComponents() local
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc602 auto rhs_type = DropRefType(std::get<1>(types)); in BroadcastCompatible() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc973 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/tensorflow/ir/
Dtf_ops_a_m.cc1268 auto rhs_type = RankedTensorType::get(rhs_dims, ranked.getElementType()); in GetHoistParams() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3390 auto rhs_type = rhs.getType().cast<RankedTensorType>(); in BroadcastBatchMatMulV2Operands() local
3443 auto rhs_type = rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/external/libabigail/src/
Dabg-ir.cc21685 const type_base& rhs_type) in maybe_propagate_canonical_type()