Home
last modified time | relevance | path

Searched refs:IsIntegralType (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dprimitive_util.h147 bool IsIntegralType(PrimitiveType type);
Dprimitive_util.cc92 bool IsIntegralType(PrimitiveType type) { in IsIntegralType() function
Dshape_util.cc433 return primitive_util::IsIntegralType(shape.element_type()); in ElementIsIntegral()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc94 (primitive_util::IsIntegralType(type) || type == PRED) in CreateScalarIdentityWithZeroComputation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc62 if (primitive_util::IsIntegralType( in HeuristicLayoutAssignment()
Dcudnn_pad_for_convolutions.cc289 if (!primitive_util::IsIntegralType(input_shape.element_type())) { in TryResolvePaddedShapesForIntegerConvolution()
Dgpu_conv_rewriter.cc639 if (primitive_util::IsIntegralType( in CreateCustomCallHelper()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc384 primitive_util::IsIntegralType(primitive_type)); in EmitTiledLlvmIrGemv()
1025 primitive_util::IsIntegralType(element_type))) { in GetDotImplementationStrategy()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter.cc254 CHECK(primitive_util::IsIntegralType(from_type) || from_type == PRED) in EmitIntegerUnaryOp()
265 if (primitive_util::IsIntegralType(to_type)) { in EmitIntegerUnaryOp()
299 CHECK(primitive_util::IsIntegralType(from_type)); in EmitIntegerUnaryOp()
352 } else if (primitive_util::IsIntegralType(type)) { in EmitIntegerUnaryOp()
1742 } else if (primitive_util::IsIntegralType(prim_type)) { in EmitElementalClamp()
2378 if (primitive_util::IsIntegralType(component_element_type)) { in MakeElementGenerator()
Dshape_inference.cc225 if (primitive_util::IsIntegralType(from_type) != in MaybeUpcast()
226 primitive_util::IsIntegralType(*preferred_element_type)) { in MaybeUpcast()
357 !primitive_util::IsIntegralType(shape.element_type())) { in InferUnaryOpShape()
1020 !primitive_util::IsIntegralType(lhs.element_type())) { in InferBinaryOpShape()
Dhlo_verifier.cc479 !primitive_util::IsIntegralType(element_type) && in HandleRng()
622 if (!primitive_util::IsIntegralType(primitive_type) && in HandleIota()
Dwhile_loop_simplifier.cc981 CHECK(primitive_util::IsIntegralType(elem_ty)) << PrimitiveType_Name(elem_ty); in TryMergeInductionVariables()
Dhlo_evaluator_typed_visitor.h1615 primitive_util::IsIntegralType(start_indices->shape().element_type()));
1678 primitive_util::IsIntegralType(start_indices->shape().element_type()));
Dalgebraic_simplifier.cc2705 primitive_util::IsIntegralType(multiply->shape().element_type()) && in HandleMultiply()
2711 primitive_util::IsIntegralType(multiply->shape().element_type()) && in HandleMultiply()
3236 primitive_util::IsIntegralType(lhs->shape().element_type())) { in HandleCompare()
Dhlo_parser.cc3257 } else if (primitive_util::IsIntegralType(shape.element_type()) || in ParseDenseLiteral()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dimage_resize_ops.cc524 if (is_kernel_bilinear || (xla::primitive_util::IsIntegralType(input_type))) { in GeneralCompile()