Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbitcast_decomposer.cc91 PrimitiveType elem_ty = instr->shape().element_type(); in Run() local
Dwhile_loop_simplifier.cc1127 HloInstruction* while_op, PrimitiveType elem_ty) { in TryMergeInductionVariables()
Dhlo_evaluator.cc1184 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/pytorch/torch/csrc/jit/ir/
Dir.cpp1914 int elem_ty = 0; in insertToList() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1444 const PrimitiveType elem_ty = b_shape.element_type(); in EmitTriangularSolveCustomCall() local
4219 absl::string_view buffer_name) { in EmitHlo021Tile()