Home
last modified time | relevance | path

Searched defs:lhs_type (Results 1 – 22 of 22) 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.cc57 auto lhs_type = input_lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
Dunroll_batch_matmul.cc159 auto lhs_type = input_lhs.getType().cast<RankedTensorType>(); in matchAndRewrite() local
Deinsum.cc505 RankedTensorType lhs_type = lhs->getType().cast<RankedTensorType>(); in reshapeForBatchMatmul() local
Dlegalize_hlo.cc948 auto lhs_type = conv_op.lhs().getType().cast<ShapedType>(); in MatchResizeOp() local
1675 auto lhs_type = lhs.getType().cast<ShapedType>(); in ConvertDot() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_arith_ops_folder.h61 auto lhs_type = arithmetic_op.x().getType().template cast<ShapedType>(); in IdentityArithmeticOpFolder() local
Dtf_ops_a_m.cc1348 auto lhs_type = RankedTensorType::get(lhs_dims, ranked.getElementType()); in GetHoistParams() local
/external/pytorch/aten/src/ATen/core/
Dunion_type.cpp306 [&](TypePtr lhs_type) { in equals()
353 [&](const TypePtr& lhs_type) -> bool { in isSubtypeOfExt()
/external/angle/third_party/spirv-tools/src/source/fuzz/
Dfuzzer_pass_add_equation_instructions.cpp221 auto lhs_type = in Apply() local
/external/swiftshader/third_party/SPIRV-Tools/source/fuzz/
Dfuzzer_pass_add_equation_instructions.cpp221 auto lhs_type = in Apply() local
/external/deqp-deps/SPIRV-Tools/source/fuzz/
Dfuzzer_pass_add_equation_instructions.cpp221 auto lhs_type = in Apply() local
/external/sdv/vsomeip/third_party/boost/spirit/include/boost/spirit/home/x3/nonterminal/
Drule.hpp44 typedef rule<ID, Attribute, force_attribute_> lhs_type; typedef
/external/trusty/arm-trusted-firmware/plat/common/
Dubsan.c39 struct type_descriptor *lhs_type; member
/external/arm-trusted-firmware/plat/common/
Dubsan.c39 struct type_descriptor *lhs_type; member
/external/coreboot/src/lib/
Dubsan.c181 struct ubsan_type_descriptor *lhs_type; member
/external/tensorflow/tensorflow/core/ir/types/
Ddialect.cc662 auto lhs_type = DropRefType(std::get<0>(types)); in BroadcastCompatible() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc982 llvm::Type* lhs_type = lhs->getType(); in GetGemvBuffersWithCanonicalType() local
/external/pytorch/torch/csrc/jit/passes/
Dshape_analysis.cpp1960 auto lhs_type = tensor_types.at(0); in PropagateCompleteShapeOnNode() local
/external/pytorch/torch/csrc/jit/frontend/
Dir_emitter.cpp2222 for (const TypePtr& lhs_type : lhs_types) { in emitIsInstance() local
4723 auto const& lhs_type = lhs->type(); in emitSimpleExpr() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc3187 auto lhs_type = lhs.getType().cast<RankedTensorType>(); in BroadcastBatchMatMulV2Operands() local
3240 auto lhs_type = lhs.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local