Home
last modified time | relevance | path

Searched defs:lhs_type (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_standard.cc43 auto lhs_type = lhs.getType().cast<TensorType>(); in matchAndRewrite() local
80 auto lhs_type = lhs.getType().cast<TensorType>(); in matchAndRewrite() local
Dlhlo_legalize_to_affine.cc51 MemRefType lhs_type = lhs.getType().cast<MemRefType>(); in matchAndRewrite() local
488 const auto& lhs_type = lhs.getType().template cast<MemRefType>(); in matchAndRewrite() local
Dchlo_legalize_to_hlo.cc1319 auto lhs_type = in matchAndRewrite() local
1369 auto lhs_type = lhs.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.cc56 auto lhs_type = input_lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
Dunroll_batch_matmul.cc218 auto lhs_type = input_lhs.getType().cast<RankedTensorType>(); in matchAndRewrite() local
Deinsum.cc397 RankedTensorType lhs_type = lhs->getType().cast<RankedTensorType>(); in reshapeForBatchMatmul() local
Dlegalize_hlo.cc749 auto lhs_type = lhs.getType().cast<ShapedType>(); in ConvertDot() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/
Dbroadcast_utils.cc33 RankedTensorType lhs_type = lhs.getType().dyn_cast<RankedTensorType>(); in IsLegalNumpyRankedBroadcast() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dchlo_ops.cc135 ShapedType lhs_type = operands[0].getType().dyn_cast<ShapedType>(); in InferBroadcastBinaryOpReturnTypeComponents() local
196 ShapedType lhs_type = operands[0].getType().dyn_cast<ShapedType>(); in inferReturnTypeComponents() local
/external/swiftshader/third_party/SPIRV-Tools/source/fuzz/
Dfuzzer_pass_add_equation_instructions.cpp217 auto lhs_type = in Apply() local
/external/angle/third_party/vulkan-deps/spirv-tools/src/source/fuzz/
Dfuzzer_pass_add_equation_instructions.cpp217 auto lhs_type = in Apply() local
/external/deqp-deps/SPIRV-Tools/source/fuzz/
Dfuzzer_pass_add_equation_instructions.cpp217 auto lhs_type = in Apply() local
/external/arm-trusted-firmware/plat/common/
Dubsan.c39 struct type_descriptor *lhs_type; member
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc601 auto lhs_type = DropRefType(std::get<0>(types)); in BroadcastCompatible() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc972 llvm::Type* lhs_type = lhs->getType(); in GetGemvBuffersWithCanonicalType() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1278 auto lhs_type = RankedTensorType::get(lhs_dims, ranked.getElementType()); in GetHoistParams() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3389 auto lhs_type = lhs.getType().cast<RankedTensorType>(); in BroadcastBatchMatMulV2Operands() local
3442 auto lhs_type = lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
/external/libabigail/src/
Dabg-ir.cc21684 maybe_propagate_canonical_type(const type_base& lhs_type, in maybe_propagate_canonical_type()