Home
last modified time | relevance | path

Searched defs:elem_ty (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc200 llvm::Type* elem_ty, absl::string_view buffer_name) { in GetSharedMemoryBufferForElementType()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath.cc39 PrimitiveType elem_ty = shape.element_type(); in DoWithUpcastToF32() local
58 auto elem_ty = shape.element_type(); in EnsureOperandIsRealFp() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc768 HloInstruction* while_op, PrimitiveType elem_ty) { in TryMergeInductionVariables()
Dhlo_evaluator.cc493 auto elem_ty = operand->shape().element_type(); in HandleIsFinite() local