Searched refs:TF_TYPE (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_test.cc | 155 #define BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, R, C) \ argument
163 test::Benchmark(#DEVICE, BiasAdd<C_TYPE>(rows, cols, TF_TYPE)).Run(iters); \
168 #define BM_BIAS_ADD_ALL(DEVICE, C_TYPE, TF_TYPE) \ argument
169 BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, 512, 2048); \
170 BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, 512, 4096); \
171 BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, 2048, 512); \
172 BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, 4096, 512);
206 #define BM_BIAS_ADD_GRAD(DEVICE, FMT, C_TYPE, TF_TYPE, R, C, CH) \ argument
216 TF_TYPE, FORMAT_##FMT)) \
222 #define BM_BIAS_ADD_GRAD_ALL(DEVICE, FORMAT, C_TYPE, TF_TYPE) \ argument
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_util.cc | 327 #define HANDLE_COMPRESS_CASE(TF_TYPE) \ argument
328 case TF_TYPE: \
330 EnumToDataType<TF_TYPE>::Type>(min_num_elements, \
|