HomeSort by relevance Sort by last modified time
    Searched refs:primitive_util (Results 1 - 25 of 30) sorted by null

1 2

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
target_machine_features.h 21 #include "tensorflow/compiler/xla/primitive_util.h"
61 (primitive_util::BitWidth(type) / 8);
dot_op_emitter.cc 569 if (!primitive_util::IsFloatingPointType(primitive_type) &&
570 !primitive_util::IsIntegralType(primitive_type)) {
    [all...]
vector_support_library.h 23 #include "tensorflow/compiler/xla/primitive_util.h"
223 return primitive_util::BitWidth(primitive_type_) / 8;
  /external/tensorflow/tensorflow/compiler/tf2xla/kernels/
cast_op.cc 20 #include "tensorflow/compiler/xla/primitive_util.h"
44 } else if (xla::primitive_util::IsComplexType(src_type_) &&
45 !xla::primitive_util::IsComplexType(dst_type_)) {
  /external/tensorflow/tensorflow/compiler/xla/tests/
dynamic_ops_test.cc 133 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
137 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
159 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
163 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
185 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
189 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
351 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
355 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
359 ->Convert(primitive_util::NativeToPrimitiveType<DataT>())
382 ->Convert(primitive_util::NativeToPrimitiveType<DataT>()
    [all...]
literal_test_util.h 213 typename T = typename primitive_util::PrimitiveTypeToNative<type>::type>
224 typename T = typename primitive_util::PrimitiveTypeToNative<type>::type>
234 typename T = typename primitive_util::PrimitiveTypeToNative<type>::type>
351 using NativeT = typename primitive_util::PrimitiveTypeToNative<type>::type;
365 using NativeT = typename primitive_util::PrimitiveTypeToNative<type>::type;
dot_operation_test.cc 23 #include "tensorflow/compiler/xla/primitive_util.h"
190 auto prim_type = primitive_util::NativeToPrimitiveType<Element>();
262 auto prim_type = primitive_util::NativeToPrimitiveType<float>();
399 auto prim_type = primitive_util::NativeToPrimitiveType<Element>();
460 auto prim_type = primitive_util::NativeToPrimitiveType<complex64>();
590 auto prim_type = primitive_util::NativeToPrimitiveType<float>();
617 auto prim_type = primitive_util::NativeToPrimitiveType<float>();
660 auto prim_type = primitive_util::NativeToPrimitiveType<float>();
matrix_ops_simple_test.cc 175 auto prim_type = primitive_util::NativeToPrimitiveType<float>();
241 auto prim_type = primitive_util::NativeToPrimitiveType<bfloat16>();
test_utils.cc 17 #include "tensorflow/compiler/xla/primitive_util.h"
30 primitive_util::NativeToPrimitiveType<FloatT>());
76 primitive_util::NativeToPrimitiveType<IntT>());
unary_op_test.cc 44 if (primitive_util::NativeToPrimitiveType<T>() == C64) {
prng_test.cc 22 #include "tensorflow/compiler/xla/primitive_util.h"
58 ShapeUtil::MakeShape(primitive_util::NativeToPrimitiveType<T>(), dims));
literal_test_util.cc 115 primitive_util::NativeToPrimitiveType<FromNativeT>()) {
117 primitive_util::NativeToPrimitiveType<ToNativeT>());
129 primitive_util::NativeToPrimitiveType<FromNativeT>()) {
fusion_test.cc 31 #include "tensorflow/compiler/xla/primitive_util.h"
90 auto prim_type = primitive_util::NativeToPrimitiveType<T>();
123 if (primitive_util::IsFloatingPointType(prim_type)) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
elemental_ir_emitter.cc 28 #include "tensorflow/compiler/xla/primitive_util.h"
167 /*exponent_bits=*/primitive_util::kBFloat16ExponentBits,
168 /*mantissa_bits=*/primitive_util::kBFloat16MantissaBits, ir_builder);
189 if (primitive_util::IsSignedIntegralType(from_type)) {
193 CHECK(primitive_util::IsUnsignedIntegralType(from_type) ||
222 CHECK(primitive_util::IsIntegralType(from_type) || from_type == PRED);
226 if (primitive_util::IsIntegralType(to_type)) {
229 primitive_util::IsSignedIntegralType(to_type));
231 if (primitive_util::IsFloatingPointType(to_type)) {
241 if (primitive_util::IsComplexType(to_type))
    [all...]
hlo_constant_folding_test.cc 195 using NativeT = typename primitive_util::PrimitiveTypeToNative<F32>::type;
shape_inference.cc 353 arg, primitive_util::ComplexComponentType(arg.element_type()));
364 !primitive_util::IsIntegralType(arg.element_type())) {
452 if (primitive_util::IsComplexType(old_element_type) &&
453 !primitive_util::IsComplexType(new_element_type)) {
475 if (primitive_util::IsComplexType(old_element_type) !=
476 primitive_util::IsComplexType(new_element_type)) {
491 if (primitive_util::BitWidth(old_element_type) !=
492 primitive_util::BitWidth(new_element_type)) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
primitive_util.cc 16 #include "tensorflow/compiler/xla/primitive_util.h"
21 namespace primitive_util { namespace in namespace:xla
88 } // namespace primitive_util
primitive_util.h 27 namespace primitive_util { namespace in namespace:xla
221 } // namespace primitive_util
literal_util.h 35 #include "tensorflow/compiler/xla/primitive_util.h"
98 // given ShapeIndex is not array. See primitive_util.h for the mapping from
    [all...]
shape_util.h 26 #include "tensorflow/compiler/xla/primitive_util.h"
237 return primitive_util::BitWidth(a.element_type()) <
238 primitive_util::BitWidth(b.element_type())
shape_util.cc 27 #include "tensorflow/compiler/xla/primitive_util.h"
270 return primitive_util::IsIntegralType(shape.element_type());
282 return primitive_util::BitWidth(shape.element_type()) == bits;
313 return primitive_util::IsComplexType(shape.element_type());
317 return primitive_util::IsFloatingPointType(shape.element_type());
362 return ChangeElementType(complex_shape, primitive_util::ComplexComponentType(
    [all...]
literal_util.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.cc 27 #include "tensorflow/compiler/xla/primitive_util.h"
203 if (primitive_util::IsIntegralType(element_type)) {
214 primitive_util::IsIntegralType(element_type)) {
216 auto opcode = primitive_util::IsSignedIntegralType(element_type)
225 primitive_util::IsIntegralType(element_type)) {
227 auto opcode = primitive_util::IsSignedIntegralType(element_type)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
computation_builder.h     [all...]

Completed in 639 milliseconds

1 2