Home
last modified time | relevance | path

Searched refs:NUM_ELEM (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dtensor_cord_test.cc253 #define CreateBM(NUM_ELEM, STRING_SIZE) \ argument
254 void BM_TensorCopyFromTensor_NumElem_##NUM_ELEM##_StringSize_##STRING_SIZE( \
256 TensorCopyFromTensorBenchmark(state, NUM_ELEM, STRING_SIZE); \
259 BM_TensorCopyFromTensor_NumElem_##NUM_ELEM##_StringSize_##STRING_SIZE); \
260 void BM_TensorCordFromTensor_NumElem_##NUM_ELEM##_StringSize_##STRING_SIZE( \
262 TensorCordFromTensorBenchmark(state, NUM_ELEM, STRING_SIZE); \
265 BM_TensorCordFromTensor_NumElem_##NUM_ELEM##_StringSize_##STRING_SIZE); \
267 BM_TensorCordFromAbslCord_NumElem_##NUM_ELEM##_StringSize_##STRING_SIZE( \
269 TensorCordFromAbslCordBenchmark(state, NUM_ELEM, STRING_SIZE); \
272 BM_TensorCordFromAbslCord_NumElem_##NUM_ELEM##_StringSize_##STRING_SIZE);
[all …]
Dsparse_matmul_op.cc1191 template <int NUM_ELEM = -1>
1196 const int num = (NUM_ELEM == -1) ? num_elements : NUM_ELEM; in CopyAndMayBeInterleaveBfloat16()
/external/grpc-grpc/test/core/iomgr/
Dbuffer_list_test.cc46 #define NUM_ELEM 5 in TestShutdownFlushesList() macro
47 gpr_atm verifier_called[NUM_ELEM]; in TestShutdownFlushesList()
48 for (auto i = 0; i < NUM_ELEM; i++) { in TestShutdownFlushesList()
55 for (auto i = 0; i < NUM_ELEM; i++) { in TestShutdownFlushesList()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/
Dprint_memrefs.mlir33 // CHECK: [[NUM_ELEM:%.*]] = alloca() : memref<1xindex>
34 // CHECK: store {{%.*}}, [[NUM_ELEM]]
36 // CHECK: [[NUM_ELEM_I64:%.*]] = index_cast [[NUM_ELEM]]
/external/arm-trusted-firmware/tools/encrypt_fw/src/
Dmain.c23 #define NUM_ELEM(x) ((sizeof(x)) / (sizeof(x[0]))) macro
77 for (i = 0 ; i < NUM_ELEM(key_algs_str) ; i++) { in get_key_alg()
150 for (i = 0; i < NUM_ELEM(common_cmd_opt); i++) { in main()
/external/arm-trusted-firmware/tools/cert_create/src/
Dmain.c58 #define NUM_ELEM(x) ((sizeof(x)) / (sizeof(x[0]))) macro
141 for (i = 0 ; i < NUM_ELEM(key_algs_str) ; i++) { in get_key_alg()
166 for (i = 0 ; i < NUM_ELEM(hash_algs_str) ; i++) { in get_hash_alg()
320 for (i = 0; i < NUM_ELEM(common_cmd_opt); i++) { in main()