Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dresult_caster_test.cc34 PrimitiveType lhs_type, rhs_type, result_type; in TEST_P() local
Doperand_upcaster_test.cc40 PrimitiveType lhs_type, rhs_type, result_type; in TEST_P() local
Dshape_inference_test.cc647 ConvolveArgs MakeConvolveArgs(PrimitiveType lhs_type, PrimitiveType rhs_type) { in MakeConvolveArgs()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbatchmatmul_to_einsum.cc58 auto rhs_type = input_rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
Dunroll_batch_matmul.cc160 auto rhs_type = input_rhs.getType().cast<RankedTensorType>(); in matchAndRewrite() local
Deinsum.cc506 RankedTensorType rhs_type = rhs->getType().cast<RankedTensorType>(); in reshapeForBatchMatmul() local
Dlegalize_hlo.cc1676 auto rhs_type = rhs.getType().cast<ShapedType>(); in ConvertDot() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_arith_ops_folder.h62 auto rhs_type = arithmetic_op.y().getType().template cast<ShapedType>(); in IdentityArithmeticOpFolder() local
Dtf_ops_a_m.cc1338 auto rhs_type = RankedTensorType::get(rhs_dims, ranked.getElementType()); in GetHoistParams() local
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dconvert_tf_quant_ops_to_mhlo.cc126 Type rhs_type = getSameShapeTensorType( in matchAndRewrite() local
/external/arm-trusted-firmware/plat/common/
Dubsan.c40 struct type_descriptor *rhs_type; member
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc663 auto rhs_type = DropRefType(std::get<1>(types)); in BroadcastCompatible() local
/external/mesa3d/src/compiler/glsl/
Dast_function.cpp1528 const glsl_type *rhs_type = in emit_inline_vector_constructor() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc983 llvm::Type* rhs_type = rhs->getType(); in GetGemvBuffersWithCanonicalType() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3188 auto rhs_type = rhs.getType().cast<RankedTensorType>(); in BroadcastBatchMatMulV2Operands() local
3241 auto rhs_type = rhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/external/libabigail/src/
Dabg-ir.cc21720 const type_base& rhs_type) in maybe_propagate_canonical_type()