Home
last modified time | relevance | path

Searched refs:NativeToPrimitiveType (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dprimitive_util.h46 PrimitiveType NativeToPrimitiveType() { in NativeToPrimitiveType() function
60 inline PrimitiveType NativeToPrimitiveType<bool>() {
66 inline PrimitiveType NativeToPrimitiveType<uint8>() {
71 inline PrimitiveType NativeToPrimitiveType<uint16>() {
76 inline PrimitiveType NativeToPrimitiveType<uint32>() {
81 inline PrimitiveType NativeToPrimitiveType<uint64>() {
87 inline PrimitiveType NativeToPrimitiveType<int8>() {
92 inline PrimitiveType NativeToPrimitiveType<int16>() {
97 inline PrimitiveType NativeToPrimitiveType<int32>() {
102 inline PrimitiveType NativeToPrimitiveType<int64>() {
[all …]
Dliteral_util.h316 primitive_util::NativeToPrimitiveType<NativeT>(), {})); in CreateR0()
324 ShapeUtil::MakeShape(primitive_util::NativeToPrimitiveType<NativeT>(), in CreateR1()
335 primitive_util::NativeToPrimitiveType<NativeT>(), in CreateR2WithLayout()
421 primitive_util::NativeToPrimitiveType<NativeT>(), dimensions, in CreateSparse()
439 primitive_util::NativeToPrimitiveType<NativeT>(), values.dimensions(), in CreateFromArrayWithLayout()
555 primitive_util::NativeToPrimitiveType<NativeT>(), dimensions)); in CreateFullWithDescendingLayout()
Dliteral.h872 primitive_util::NativeToPrimitiveType<NativeT>()) in data()
874 << PrimitiveType_Name(primitive_util::NativeToPrimitiveType<NativeT>()) in data()
885 primitive_util::NativeToPrimitiveType<NativeT>()) in data()
887 << PrimitiveType_Name(primitive_util::NativeToPrimitiveType<NativeT>()) in data()
996 primitive_util::NativeToPrimitiveType<NativeT>()); in PopulateR1()
1007 primitive_util::NativeToPrimitiveType<NativeT>()); in PopulateR2()
1030 primitive_util::NativeToPrimitiveType<NativeT>()); in PopulateFromArray()
1087 primitive_util::NativeToPrimitiveType<NativeT>()); in PopulateInternal()
1138 primitive_util::NativeToPrimitiveType<NativeT>()); in PopulateWithValue()
Dliteral_util.cc53 primitive_util::NativeToPrimitiveType<FromNativeT>()) { in ConvertType()
55 primitive_util::NativeToPrimitiveType<ToNativeT>()); in ConvertType()
67 primitive_util::NativeToPrimitiveType<FromNativeT>()) { in ConvertType()
Dshape_util.h393 return ShapeUtil::MakeShape(primitive_util::NativeToPrimitiveType<T>(), in MakeShapeWithType()
Dliteral_test.cc974 primitive_util::NativeToPrimitiveType<uint32>(), dimensions, layout); in TEST_F()
1180 primitive_util::NativeToPrimitiveType<uint32>(), data.dimensions, in TEST_F()
1222 primitive_util::NativeToPrimitiveType<uint32>(), data.dimensions, in TEST_F()
Dliteral.cc1240 primitive_util::NativeToPrimitiveType<NativeDestT>())); in ConvertBetweenNativeTypesWithConverter()
/external/tensorflow/tensorflow/compiler/xla/tests/
Ddynamic_ops_test.cc128 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR1()
132 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR1()
154 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR2()
158 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR2()
189 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR3()
193 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR3()
380 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR0()
384 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR0()
388 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR0()
407 .Convert(primitive_util::NativeToPrimitiveType<DataT>()) in RunR1()
[all …]
Dtest_utils.cc107 primitive_util::NativeToPrimitiveType<FloatT>()); in PopulateWithFloatingPointData()
121 primitive_util::NativeToPrimitiveType<ComplexT>()); in PopulateWithComplexData()
123 result->shape(), primitive_util::NativeToPrimitiveType<InnerFloatT>()); in PopulateWithComplexData()
146 primitive_util::NativeToPrimitiveType<half>()); in PopulateWithFloatingPointData()
160 primitive_util::NativeToPrimitiveType<bfloat16>()); in PopulateWithFloatingPointData()
173 primitive_util::NativeToPrimitiveType<IntT>()); in PopulateWithRandomIntegralData()
278 primitive_util::NativeToPrimitiveType<IntT>()); in PopulateWithRandomIntegralDataWithBounds()
Ddot_operation_test.cc236 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestImpl()
318 auto prim_type = primitive_util::NativeToPrimitiveType<NativeT>(); in TestImpl()
485 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestImpl()
516 auto prim_type = primitive_util::NativeToPrimitiveType<complex64>(); in XLA_TEST_F()
785 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in XLA_TYPED_TEST()
816 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in XLA_TYPED_TEST()
Dunary_op_test.cc44 if (primitive_util::NativeToPrimitiveType<T>() == C64) { in AbsSize0TestHelper()
Dmatrix_ops_simple_test.cc195 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestImpl()
Dprng_test.cc56 ShapeUtil::MakeShape(primitive_util::NativeToPrimitiveType<T>(), dims)); in UniformTest()
Dreduce_test.cc288 xla::primitive_util::NativeToPrimitiveType<NativeT>(), {rows, cols}); in RunVectorizedReduceTestForType()
893 primitive_util::NativeToPrimitiveType<T>(), &builder); in DoTest()
Dfusion_test.cc94 auto prim_type = primitive_util::NativeToPrimitiveType<T>(); in TestElementwise2D()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dsequence_ops.cc64 xla::Iota(builder, xla::primitive_util::NativeToPrimitiveType<T>(), in CreateRangeTensor()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc148 CHECK_EQ(primitive_util::NativeToPrimitiveType<T>(), in RunCudnnConvImpl()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser.cc2449 primitive_util::NativeToPrimitiveType<LiteralNativeT>(); in CheckParsedValueIsInRange()
2516 primitive_util::NativeToPrimitiveType<LiteralNativeT>()), in CheckParsedValueIsInRange()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h1937 primitive_util::NativeToPrimitiveType<NativeT>(), {length})); in ConstantR1()