HomeSort by relevance Sort by last modified time
    Searched refs:elem_ty (Results 1 - 5 of 5) 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/
while_loop_simplifier.cc 744 // are effective-scalars of type `elem_ty`. Some terminology:
766 // each integral type elem_ty.
768 HloInstruction* while_op, PrimitiveType elem_ty) {
769 CHECK(primitive_util::IsIntegralType(elem_ty)) << PrimitiveType_Name(elem_ty);
    [all...]
hlo_evaluator.cc 493 auto elem_ty = operand->shape().element_type(); local
494 switch (elem_ty) {
517 PrimitiveType_Name(elem_ty));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
kernel_tiling.h 179 llvm::Type* elem_ty, absl::string_view buffer_name);
kernel_tiling.cc 200 llvm::Type* elem_ty, absl::string_view buffer_name) {
211 llvm::ArrayType::get(elem_ty, GetTileSizeForDimension(DimX) + 1),

Completed in 457 milliseconds