Searched refs:elem_ty (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | while_loop_simplifier.cc | 980 HloInstruction* while_op, PrimitiveType elem_ty) { in TryMergeInductionVariables() argument
981 CHECK(primitive_util::IsIntegralType(elem_ty)) << PrimitiveType_Name(elem_ty); in TryMergeInductionVariables()
1008 .WithShape(m::Shape().WithElementType(elem_ty)))) { in TryMergeInductionVariables()
1071 ShapeUtil::MakeShape(elem_ty, /*dimensions=*/{}); in TryMergeInductionVariables()
1118 HloInstruction::CreateConstant(LiteralUtil::One(elem_ty))))); in TryMergeInductionVariables()
1136 HloInstruction::CreateConstant(LiteralUtil::Zero(elem_ty)))); in TryMergeInductionVariables()
1277 for (auto elem_ty : {S8, U8, S32, U32, S64, U64}) { in Run()
1279 TryMergeInductionVariables(while_op, elem_ty)); in Run()
|
| D | hlo_evaluator.cc | 530 auto elem_ty = operand->shape().element_type(); in HandleIsFinite() local
531 switch (elem_ty) { in HandleIsFinite()
554 PrimitiveType_Name(elem_ty)); in HandleIsFinite()
|
| /external/llvm-project/lldb/unittests/DataFormatter/ |
| D | StringPrinterTests.cpp | 29 template <StringPrinter::StringElementType elem_ty>
43 const bool success = StringPrinter::ReadBufferAndDumpToStream<elem_ty>(opts); in format()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 71 PrimitiveType elem_ty = shape.element_type(); in DoWithUpcastToF32() local
72 bool needs_upcast = absl::c_linear_search(upcast_types, elem_ty); in DoWithUpcastToF32()
79 result = ConvertElementType(result, elem_ty); in DoWithUpcastToF32()
90 auto elem_ty = shape.element_type(); in EnsureOperandIsRealFp() local
91 if (!primitive_util::IsFloatingPointType(elem_ty)) { in EnsureOperandIsRealFp()
94 op_name, PrimitiveType_Name(elem_ty)); in EnsureOperandIsRealFp()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 439 Type elem_ty; in Verify() local
443 elem_ty = idx_tensor.getElementType(); in Verify()
445 if (first_elem_ty != elem_ty) { in Verify()
448 << first_elem_ty << " vs " << elem_ty << ")"; in Verify()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 4846 auto get_shared_memory_buffer = [&](llvm::Type* elem_ty, in EmitHlo021Tile()
4855 llvm::ArrayType::get(elem_ty, mapping_scheme.GetTileSizeX() + 1), in EmitHlo021Tile()
|