| /external/tensorflow/tensorflow/core/kernels/ |
| D | cwise_ops_test.cc | 45 #define BM_UNARY(DEVICE, FUNC, T, TYPE) \ argument
108 #define BM_BINARY_SCALAR(DEVICE, FUNC) \ argument
183 #define BM_CUBE(DEVICE, Impl) \ argument
224 #define BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, R, C) \ argument
241 #define BM_BIAS_ADD_ALL(DEVICE, C_TYPE, TF_TYPE) \ argument
279 #define BM_BIAS_ADD_GRAD(DEVICE, FMT, C_TYPE, TF_TYPE, R, C, CH) \ argument
300 #define BM_BIAS_ADD_GRAD_ALL(DEVICE, FORMAT, C_TYPE, TF_TYPE) \ argument
348 #define BM_BCAST_ADD_ROW(DEVICE, R, C) \ argument
364 #define BM_BCAST_ADD_ROW_ALL(DEVICE) \ argument
376 #define BM_BCAST_ADD_COL(DEVICE, R, C) \ argument
[all …]
|
| D | random_binomial_op_test.cc | 70 #define BM_RandomBinomialInv(DEVICE, B, S) \ argument
80 #define BM_RandomBinomialRej(DEVICE, B, S) \ argument
90 #define BM_RandomBinomialInvComplement(DEVICE, B, S) \ argument
100 #define BM_RandomBinomialRejComplement(DEVICE, B, S) \ argument
|
| D | parameterized_truncated_normal_op_test.cc | 110 #define BM_PTruncatedNormalDev(DEVICE, B, S) \ argument
120 #define BM_PTruncatedNormalDev_2SD(DEVICE, B, S) \ argument
130 #define BM_PTruncatedNormalDev_OneTail(DEVICE, B, S) \ argument
|
| D | strided_slice_op_impl.h | 243 #define INSTANTIATE_DIM1_AND_UP_HANDLERS(DEVICE, T, DIM) \ argument
257 #define INSTANTIATE_DIM0_AND_UP_HANDLERS(DEVICE, T, DIM) \ argument
262 #define INSTANTIATE(DEVICE, T, DIM) \ argument
265 #define INSTANTIATE(DEVICE, T, DIM) \ argument
|
| D | in_topk_op_test.cc | 65 #define BM_NAME(T, TARGETS, CLASSES, K, DEVICE) \ argument
68 #define BM_InTopK(T, TARGETS, CLASSES, K, DEVICE) \ argument
|
| D | bias_op_test.cc | 46 #define BM_BiasAddNHWC(N, W, H, C, DEVICE) \ argument
56 #define BM_BiasAddGradNHWC(N, W, H, C, DEVICE) \ argument
|
| D | fused_batch_norm_op_test.cc | 280 #define BM_NAME(NAME, N, H, W, C, T, IT, FORMAT, DEVICE) \ argument
288 #define BM_FusedBatchNorm(N, H, W, C, T, IS_TRAINING, FORMAT, DEVICE) \ argument
328 #define BM_FusedBatchNormGrad(N, H, W, C, T, IS_TRAINING, FORMAT, DEVICE) \ argument
342 #define BM_FusedBatchNormGradResnetShapes(T, IS_TRAINING, FORMAT, DEVICE) \ argument
|
| D | cast_op.h | 30 #define SPECIALIZE_CAST(DEVICE, OUT_TYPE, IN_OUT) \ argument
48 #define SPECIALIZE_CAST(DEVICE, OUT_TYPE, IN_OUT) \ argument
|
| D | stateless_random_ops_v2.cc | 265 #define REGISTER(DEVICE, TYPE) \ argument
293 #define REGISTER_FULL_INT(DEVICE, TYPE) \ argument
302 #define REGISTER_INT(DEVICE, TYPE) \ argument
329 #define REGISTER_GET_KCA(DEVICE) \ argument
|
| D | stateless_random_ops.cc | 204 #define REGISTER(DEVICE, TYPE) \ argument
232 #define REGISTER_FULL_INT(DEVICE, TYPE) \ argument
241 #define REGISTER_INT(DEVICE, TYPE) \ argument
|
| D | stateful_random_ops.cc | 353 #define REGISTER_FloatOps(DEVICE, TYPE) \ argument
403 #define REGISTER_StatefulUniformInt(DEVICE, TYPE) \ argument
422 #define REGISTER_StatefulUniformFullInt(DEVICE, TYPE) \ argument
442 #define REGISTER_RngSkip(DEVICE) \ argument
|
| D | xent_op_test.cc | 36 #define BM_XentDev(BATCH, CLASS, DEVICE) \ argument
|
| D | diag_op_test.cc | 33 #define BM_DiagDev(N, T, TFTYPE, DEVICE) \ argument
|
| D | sparse_xent_op_test.cc | 44 #define BM_SparseXentDev(BATCH, CLASS, DEVICE) \ argument
|
| D | cast_op_impl.h | 56 #define CAST_CASE(DEVICE, IN, OUT) \ argument
|
| D | multinomial_op_test.cc | 43 #define BM_MultinomialDev(DEVICE, B, C, S) \ argument
|
| D | one_hot_op_test.cc | 58 #define BM_OneHot(BATCH, CLASS, AXIS, DEVICE) \ argument
|
| D | random_poisson_op_test.cc | 60 #define BM_Poisson(DEVICE, BITS, MAGNITUDE) \ argument
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_ops.h | 56 #define REGISTER_XLA_LAUNCH_KERNEL(DEVICE, KERNEL, TYPES) \ argument
63 #define REGISTER_XLA_COMPILE_KERNEL(DEVICE, KERNEL, TYPES) \ argument
72 #define REGISTER_XLA_RUN_KERNEL(DEVICE, KERNEL, TYPES) \ argument
75 #define REGISTER_XLA_DEVICE_KERNELS(DEVICE, TYPES) \ argument
|
| D | xla_ops_on_regular_devices.cc | 23 #define REGISTER_XLA_OPS_ON_DEVICE(DEVICE) \ argument
|
| /external/OpenCL-CTS/test_conformance/compiler/ |
| D | procs.h | 27 #define check_compiler_available(DEVICE) \ argument
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | adjust_contrast_op_benchmark_test.cc | 39 #define BM_AdjustContrastDev(DEVICE, B, W, H) \ argument
|
| D | mirror_pad_op_benchmark_test.cc | 42 #define BM_MirrorPadDev(DEVICE, B, W, H, D, P, MODE) \ argument
|
| D | resize_op_benchmark_test.cc | 46 #define BM_ResizeDev(DEVICE, ALGORITHM, B, W, H, C, OW, OH) \ argument
|
| D | non_max_suppression_op_benchmark_test.cc | 51 #define BM_CombinedNonMaxSuppressionDev(DEVICE, B, BN, CN, Q) \ argument
|