Home
last modified time | relevance | path

Searched defs:DEVICE (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dcwise_ops_test.cc45 #define BM_UNARY(DEVICE, FUNC, T, TYPE) \ argument
100 #define BM_BINARY_SCALAR(DEVICE, FUNC) \ argument
144 #define BM_BIAS_ADD(DEVICE, C_TYPE, TF_TYPE, R, C) \ argument
156 #define BM_BIAS_ADD_ALL(DEVICE, C_TYPE, TF_TYPE) \ argument
194 #define BM_BIAS_ADD_GRAD(DEVICE, FMT, C_TYPE, TF_TYPE, R, C, CH) \ argument
209 #define BM_BIAS_ADD_GRAD_ALL(DEVICE, FORMAT, C_TYPE, TF_TYPE) \ argument
249 #define BM_BCAST_ADD_ROW(DEVICE, R, C) \ argument
260 #define BM_BCAST_ADD_ROW_ALL(DEVICE) \ argument
275 #define BM_BCAST_ADD_COL(DEVICE, R, C) \ argument
286 #define BM_BCAST_ADD_COL_ALL(DEVICE) \ argument
Dparameterized_truncated_normal_op_test.cc110 #define BM_PTruncatedNormalDev(DEVICE, B, S) \ argument
117 #define BM_PTruncatedNormalDev_2SD(DEVICE, B, S) \ argument
124 #define BM_PTruncatedNormalDev_OneTail(DEVICE, B, S) \ argument
Dstrided_slice_op_impl.h244 #define INSTANTIATE_DIM1_AND_UP_HANDLERS(DEVICE, T, DIM) \ argument
258 #define INSTANTIATE_DIM0_AND_UP_HANDLERS(DEVICE, T, DIM) \ argument
263 #define INSTANTIATE(DEVICE, T, DIM) \ argument
266 #define INSTANTIATE(DEVICE, T, DIM) \ argument
Dxent_op_test.cc36 #define BM_XentDev(BATCH, CLASS, DEVICE) \ argument
Dadjust_contrast_op_benchmark_test.cc39 #define BM_AdjustContrastDev(DEVICE, B, W, H) \ argument
Ddiag_op_test.cc33 #define BM_DiagDev(N, T, TFTYPE, DEVICE) \ argument
Dsparse_xent_op_test.cc44 #define BM_SparseXentDev(BATCH, CLASS, DEVICE) \ argument
Dresize_op_benchmark_test.cc44 #define BM_ResizeDev(DEVICE, ALGORITHM, B, W, H) \ argument
Dmultinomial_op_test.cc43 #define BM_MultinomialDev(DEVICE, B, C, S) \ argument
Drandom_poisson_op_test.cc60 #define BM_Poisson(DEVICE, BITS, MAGNITUDE) \ argument
Dmatmul_op_test.cc36 #define BM_MatmulDev(M, K, N, TA, TB, T, TFTYPE, DEVICE) \ argument
Dbatch_matmul_op_test.cc36 #define BM_BatchMatmulDev(B, M, K, N, TA, TB, T, TFTYPE, DEVICE) \ argument
Dgather_nd_op_test.cc109 #define BM_GATHER_ND(DEVICE, INDEX) \ argument
Dspacetobatch_benchmark_test.cc59 #define BM_SpaceToBatchDev(OP, DEVICE, DTYPE, B, H, W, D, BS, P00, P01, P10, … argument
Droll_op_test.cc451 #define BM_ROLL_OUTER(DEVICE) \ argument
466 #define BM_ROLL_ALL(DEVICE) \ argument
Drandom_op_test.cc61 #define BM_RNG(DEVICE, RNG) \ argument
Dsparse_tensor_dense_matmul_op_test.cc71 #define BM_SparseTensorDenseMatmulDev(NNZ, M, K, N, TA, TB, DEVICE) \ argument
Dtranspose_functor_cpu.cc110 #define INSTANTIATE(DEVICE) \ argument
Ddynamic_partition_op_test.cc190 #define BM_DYNAMIC_PARTITION(DEVICE, T, num) \ argument
Dgather_op_test.cc209 #define BM_GATHER(DEVICE, INDEX) \ argument
Dquantize_and_dequantize_op_test.cc409 #define BM_SIMPLE_QUAN_DEQUAN(DEVICE) \ argument
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h43 #define REGISTER_XLA_LAUNCH_KERNEL(DEVICE, KERNEL, TYPES) \ argument
50 #define REGISTER_XLA_DEVICE_KERNELS(DEVICE, TYPES) \ argument
/external/vogar/src/vogar/
DModeId.java25 DEVICE, enumConstant
/external/ltp/testcases/kernel/device-drivers/zram/
Dzram03.c44 #define DEVICE "/dev/zram0" macro
/external/vulkan-validation-layers/demos/smoke/
Dgenerate-dispatch-table.py28 DEVICE = 3 variable in Command

12