Home
last modified time | relevance | path

Searched defs:tupleType (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dmhlo_flatten_tuple.cc48 auto tupleType = value.getType().cast<TupleType>(); in flattenTupleType() local
56 ValueRange flattenValues, Type tupleType) { in createTupleValue()
H A Dexpand_hlo_tuples.cc67 auto tupleType = type.dyn_cast_or_null<TupleType>(); in expandTupledTensorInReturnOp() local
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dtype_test.cpp42 const auto tupleType = TupleType::create({type, IntType::get(), type}); in TEST() local
57 if (auto tupleType = t.cast<TupleType>()) { in TEST() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dpython_print.cpp711 } else if (const auto tupleType = type->cast<TupleType>()) { in registerClassDependencies() local
1544 } else if (auto tupleType = type->cast<TupleType>()) { in printNamedType() local
H A Dimport_source.cpp778 } else if (auto tupleType = serializable_type->cast<TupleType>()) { in attr() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
H A DStablehloOps.cpp5052 auto tupleType = operands[0].getType().dyn_cast<TupleType>(); in inferReturnTypes() local
6386 TupleType tupleType = current.dyn_cast<TupleType>(); in getTypeFromTupleIndices() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/IR/
H A Dhlo_ops.cc7486 auto tupleType = operands[0].getType().dyn_cast<TupleType>(); in inferReturnTypes() local
9212 TupleType tupleType = current.dyn_cast<TupleType>(); in getTypeFromTupleIndices() local