HomeSort by relevance Sort by last modified time
    Searched defs:elem_ty (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/client/lib/
math.cc 39 PrimitiveType elem_ty = shape.element_type();
40 bool needs_upcast = absl::c_linear_search(upcast_types, elem_ty);
47 result = ConvertElementType(result, elem_ty);
58 auto elem_ty = shape.element_type(); local
59 if (!primitive_util::IsFloatingPointType(elem_ty)) {
62 op_name, PrimitiveType_Name(elem_ty));
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator.cc 493 auto elem_ty = operand->shape().element_type(); local
494 switch (elem_ty) {
517 PrimitiveType_Name(elem_ty));
    [all...]

Completed in 323 milliseconds