| /external/tensorflow/tensorflow/core/grappler/utils/ |
| grappler_test.h | 81 template <DataType DTYPE>
83 typedef typename EnumToDataType<DTYPE>::Type T;
84 Tensor tensor(DTYPE, shape);
91 template <DataType DTYPE>
94 typename EnumToDataType<DTYPE>::Type value) const {
95 typedef typename EnumToDataType<DTYPE>::Type T;
96 Tensor tensor(DTYPE, shape);
|
| /external/fec/ |
| gen_ccsds_tal.c | 16 #define DTYPE unsigned char
17 DTYPE Taltab[256],Tal1tab[256];
19 static DTYPE tal[] = { 0x8d, 0xef, 0xec, 0x86, 0xfa, 0x99, 0xaf, 0x7b };
|
| /external/tensorflow/tensorflow/core/kernels/ |
| spacetobatch_benchmark_test.cc | 26 DataType dtype, const std::vector<std::pair<int, int>>& paddings) {
37 if (dtype == DT_FLOAT) {
45 } else if (dtype == DT_HALF) {
59 #define BM_SpaceToBatchDev(OP, DEVICE, DTYPE, B, H, W, D, BS, P00, P01, P10, \
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/ |
| utils.cc | 424 #define HANDLE_DOUBLE_CASE(DTYPE) \
425 case DTYPE: \
426 if (!SafeSetDoubleScalarTensorValue<EnumToDataType<DTYPE>::Type>( \
429 " in tensor of type " #DTYPE); \
433 #define HANDLE_INT_CASE(DTYPE) \
434 case DTYPE: \
435 if (!SafeSetIntScalarTensorValue<EnumToDataType<DTYPE>::Type>(value, \
438 " in tensor of type " #DTYPE); \
442 Status SetTensorValue(DataType dtype, int value, Tensor* tensor) {
449 switch (dtype) {
[all...] |
| /external/libvpx/libvpx/third_party/libyuv/source/ |
| row_any.cc | 778 #define ANY11C(NAMEANY, ANY_SIMD, SBPP, BPP, STYPE, DTYPE, MASK)
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| constant_folding.cc | 156 Tensor axis_tensor(axis_input.dtype(), axis_input.shape());
160 *axis = axis_input.dtype() == DT_INT64
370 const DataType type = output[0].dtype();
388 (*node->mutable_attr())["dtype"].set_type(type);
421 if (raw_val.dtype() != DT_INT32) {
424 Tensor value(raw_val.dtype(), raw_val.tensor_shape());
447 const DataType type = output[port_idx].dtype();
479 (*added_node->mutable_attr())["dtype"].set_type(type);
529 if (raw_val.dtype() != DT_INT64 && raw_val.dtype() != DT_INT32)
697 DataType dtype = reduction_indices_prop.dtype(); local
1011 DataType dtype = DT_INVALID; local
1575 const auto dtype = node.attr().at("dtype").type(); local
1614 const auto dtype = node.attr().at("dtype").type(); local
1644 const DataType dtype = GetDataTypeFromNodeOrProps(*node, properties); local
1675 const DataType dtype = GetDataTypeFromNodeOrProps(*node, properties); local
1722 const DataType dtype = GetDataTypeFromNodeOrProps(*node, properties); local
[all...] |
| constant_folding_test.cc | 37 template <DataType DTYPE>
40 typedef typename EnumToDataType<DTYPE>::Type T;
42 Output x = ops::Placeholder(s.WithOpName("x"), DTYPE,
44 Output v = ops::Variable(s.WithOpName("v"), {2, 2}, DTYPE);
45 Tensor zeros_t(DTYPE, TensorShape({2, 2}));
46 Tensor ones_t(DTYPE, TensorShape({2, 2}));
47 Tensor x_t(DTYPE, TensorShape({2, 2}));
59 if (DTYPE == DT_BOOL) {
102 if (DTYPE == DT_BOOL) {
717 EXPECT_EQ(DT_FLOAT, node.attr().at("dtype").type())
[all...] |
