Home
last modified time | relevance | path

Searched refs:DTYPE (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.h81 template <DataType DTYPE>
83 typedef typename EnumToDataType<DTYPE>::Type T; in GenerateRandomTensor()
84 Tensor tensor(DTYPE, shape); in GenerateRandomTensor()
91 template <DataType DTYPE>
94 typename EnumToDataType<DTYPE>::Type value) const { in GenerateConstantTensor()
95 typedef typename EnumToDataType<DTYPE>::Type T; in GenerateConstantTensor()
96 Tensor tensor(DTYPE, shape); in GenerateConstantTensor()
/external/fec/
Dgen_ccsds_tal.c16 #define DTYPE unsigned char macro
17 DTYPE Taltab[256],Tal1tab[256];
19 static DTYPE tal[] = { 0x8d, 0xef, 0xec, 0x86, 0xfa, 0x99, 0xaf, 0x7b };
/external/tensorflow/tensorflow/core/kernels/
Dspacetobatch_benchmark_test.cc59 #define BM_SpaceToBatchDev(OP, DEVICE, DTYPE, B, H, W, D, BS, P00, P01, P10, … argument
62 …BM_##OP##_##DEVICE##_##DTYPE##_##B##_##H##_##W##_##D##_bs##BS##_pad##P00##_##P01##_##P10##_##P11( \
67 … #OP, TensorShape({B, H, W, D}), BS, DTYPE, \
72 … BM_##OP##_##DEVICE##_##DTYPE##_##B##_##H##_##W##_##D##_bs##BS##_pad##P00##_##P01##_##P10##_##P11);
/external/tensorflow/tensorflow/core/grappler/
Dutils.cc424 #define HANDLE_DOUBLE_CASE(DTYPE) \ argument
425 case DTYPE: \
426 if (!SafeSetDoubleScalarTensorValue<EnumToDataType<DTYPE>::Type>( \
429 " in tensor of type " #DTYPE); \
433 #define HANDLE_INT_CASE(DTYPE) \ argument
434 case DTYPE: \
435 if (!SafeSetIntScalarTensorValue<EnumToDataType<DTYPE>::Type>(value, \
438 " in tensor of type " #DTYPE); \
/external/libvpx/libvpx/third_party/libyuv/source/
Drow_any.cc778 #define ANY11C(NAMEANY, ANY_SIMD, SBPP, BPP, STYPE, DTYPE, MASK) \ argument
779 void NAMEANY(const STYPE* src_ptr, DTYPE* dst_ptr, int scale, int width) { \
781 SIMD_ALIGNED(DTYPE out[32]); \
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc967 #define SET_TENSOR_VAL_CASE(DTYPE, TYPE, NAME) \ argument
968 case DTYPE: \
1555 #define IS_VALUE_CASE(DTYPE, VALUE) \ argument
1556 case DTYPE: \
1557 return AllValuesAre<EnumToDataType<DTYPE>::Type>( \
1558 node.attr().at("value").tensor(), EnumToDataType<DTYPE>::Type(VALUE))
Dconstant_folding_test.cc37 template <DataType DTYPE>
40 typedef typename EnumToDataType<DTYPE>::Type T; in SimpleNeutralElementTest()
42 Output x = ops::Placeholder(s.WithOpName("x"), DTYPE, in SimpleNeutralElementTest()
44 Output v = ops::Variable(s.WithOpName("v"), {2, 2}, DTYPE); in SimpleNeutralElementTest()
45 Tensor zeros_t(DTYPE, TensorShape({2, 2})); in SimpleNeutralElementTest()
46 Tensor ones_t(DTYPE, TensorShape({2, 2})); in SimpleNeutralElementTest()
47 Tensor x_t(DTYPE, TensorShape({2, 2})); in SimpleNeutralElementTest()
59 if (DTYPE == DT_BOOL) { in SimpleNeutralElementTest()
102 if (DTYPE == DT_BOOL) { in SimpleNeutralElementTest()