Searched refs:TF_TYPE (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_test.cc | 224 #define BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, R, C) \ argument
231 test::Benchmark(#DEVICE, BiasAdd<C_TYPE>(rows, cols, TF_TYPE), \
241 #define BM_BIAS_ADD_ALL(DEVICE, C_TYPE, TF_TYPE) \ argument
242 BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, 512, 2048); \
243 BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, 512, 4096); \
244 BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, 2048, 512); \
245 BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, 4096, 512);
279 #define BM_BIAS_ADD_GRAD(DEVICE, FMT, C_TYPE, TF_TYPE, R, C, CH) \ argument
289 BiasAddGrad<C_TYPE>(rows, cols, channels, TF_TYPE, FORMAT_##FMT), \
300 #define BM_BIAS_ADD_GRAD_ALL(DEVICE, FORMAT, C_TYPE, TF_TYPE) \ argument
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor_util.cc | 334 #define HANDLE_COMPRESS_CASE(TF_TYPE) \ argument
335 case TF_TYPE: \
337 EnumToDataType<TF_TYPE>::Type>(min_num_elements, \
|