Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc980 HloInstruction* while_op, PrimitiveType elem_ty) { in TryMergeInductionVariables()
Dhlo_evaluator.cc530 auto elem_ty = operand->shape().element_type(); in HandleIsFinite() local
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath.cc71 PrimitiveType elem_ty = shape.element_type(); in DoWithUpcastToF32() local
90 auto elem_ty = shape.element_type(); in EnsureOperandIsRealFp() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/
Dhlo_ops.cc439 Type elem_ty; in Verify() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4847 absl::string_view buffer_name) { in EmitHlo021Tile()