OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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