Home
last modified time | relevance | path

Searched refs:BytesProcessed (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dcast_op_test.cc123 testing::BytesProcessed(static_cast<int64>(iters) * num * in TEST_ALL_CASTS_FROM()
132 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_gpu_float_int64()
146 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_cpu_bool_float()
155 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_gpu_bool_float()
169 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_cpu_float_bfloat16()
178 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_cpu_bfloat16_float()
187 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_cpu_float_half()
196 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_cpu_half_float()
205 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_gpu_float_half()
216 testing::BytesProcessed(static_cast<int64>(iters) * num * in BM_gpu_half_float()
Dscan_ops_test.cc73 testing::BytesProcessed(static_cast<int64>(iters) * num_x * sizeof(T)); in LargeOneDimensional()
80 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * in DoRowCumsum()
88 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * in DoColCumsum()
96 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * in Do3DYCumsum()
Dconcat_op_test.cc63 testing::BytesProcessed(static_cast<int64>(iters) * in ConcatHelper()
120 testing::BytesProcessed(static_cast<int64>(iters) * kDim1 * dim2 * in ConcatManyHelper()
140 testing::BytesProcessed(static_cast<int64>(iters) * in MemcpyAlternativeHelper()
176 testing::BytesProcessed(static_cast<int64>(iters) * kDim1 * dim2 * in MemcpyManyAlternative1()
211 testing::BytesProcessed(static_cast<int64>(iters) * kDim1 * dim2 * in MemcpyManyAlternative2()
Dreduction_ops_test.cc90 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * in ReduceToScalar()
98 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * in DoRowReduce()
106 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * in DoColReduce()
114 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * in Do3DYReduce()
122 testing::BytesProcessed(static_cast<int64>(iters) * num_x * num_y * in Do3DXZReduce()
Dcwise_ops_test.cc50 testing::BytesProcessed(tot * sizeof(T)); \
106 testing::BytesProcessed(tot * sizeof(float)); \
162 testing::BytesProcessed(tot * sizeof(C_TYPE)); \
214 testing::BytesProcessed(tot * sizeof(C_TYPE)); \
277 testing::BytesProcessed(tot * sizeof(float)); \
304 testing::BytesProcessed(tot * sizeof(float)); \
331 testing::BytesProcessed(tot * sizeof(float)); \
359 testing::BytesProcessed(tot * sizeof(float)); \
Dtraining_ops_test.cc80 testing::BytesProcessed(tot * sizeof(float)); in BM_SGD()
112 testing::BytesProcessed(tot * sizeof(float)); in BM_Adagrad()
146 testing::BytesProcessed(tot * sizeof(float)); in BM_Momentum()
189 testing::BytesProcessed(tot * sizeof(float)); in BM_Adam()
229 testing::BytesProcessed(tot * sizeof(float)); in BM_RMSProp()
266 testing::BytesProcessed(tot * sizeof(float)); in BM_AddSign()
303 testing::BytesProcessed(tot * sizeof(float)); in BM_PowerSign()
Dunique_op_test.cc78 testing::BytesProcessed(static_cast<int64>(iters) * dim * sizeof(int32)); in BM_Unique_INT32()
98 testing::BytesProcessed(static_cast<int64>(iters) * dim * 200 * in BM_Unique_INT32_Repeat()
135 testing::BytesProcessed(static_cast<int64>(iters) * dim * sizeof(string)); in BM_Unique_STRING()
Dsegment_reduction_ops_test.cc91 testing::BytesProcessed(bytes_per_iter * iters); in BM_SegmentReduction()
152 testing::BytesProcessed(static_cast<int64>(iters) * (kDim1 * kDim2) * in SparseSegmentMeanGradHelper()
Dslice_op_test.cc66 testing::BytesProcessed(static_cast<int64>(iters) * kDim * size * sizeof(T)); in SliceHelper()
Dgather_nd_op_test.cc113 testing::BytesProcessed(tot * sizeof(float)); \
Dstrided_slice_op_test.cc73 testing::BytesProcessed(static_cast<int64>(iters) * kDim * size * sizeof(T)); in SliceHelper()
Dsparse_tensor_dense_matmul_op_test.cc77 testing::BytesProcessed(static_cast<int64>(iters) * items_per_iter * \
Droll_op_test.cc458 testing::BytesProcessed(num_items * sizeof(float)); \
473 testing::BytesProcessed(num_items * sizeof(float)); \
Ddequantize_op_test.cc168 testing::BytesProcessed(iters * num_values * (sizeof(float) + sizeof(T))); in BM_DequantizeMinCombinedCpu()
Dgather_op_test.cc214 testing::BytesProcessed(tot * sizeof(float)); \
Dsparse_to_dense_op_test.cc271 tensorflow::testing::BytesProcessed(bytes_per_iter * iters); in BM_SparseToDense()
Dreverse_op_test.cc204 testing::BytesProcessed(num_items * sizeof(T)); in RunReverseRowsBenchmark()
Dquantized_concat_op_test.cc281 testing::BytesProcessed(static_cast<int64>(iters) * in ConcatHelper()
/external/tensorflow/tensorflow/core/framework/
Dbfloat16_test.cc158 testing::BytesProcessed(tot * (sizeof(float) + sizeof(bfloat16))); in BM_FloatToBFloat16()
177 testing::BytesProcessed(tot * (sizeof(float) + sizeof(bfloat16))); in BM_BFloat16ToFloat()
/external/tensorflow/tensorflow/core/platform/
Dtest_benchmark.h109 void BytesProcessed(int64);
/external/tensorflow/tensorflow/core/lib/hash/
Dcrc32c_test.cc79 testing::BytesProcessed(static_cast<int64>(iters) * len); in BM_CRC()
Dhash_test.cc81 testing::BytesProcessed(static_cast<int64>(iters) * len); in BM_Hash32()
/external/tensorflow/tensorflow/core/platform/default/
Dtest_benchmark.cc249 void BytesProcessed(int64 n) { bytes_processed = n; } in BytesProcessed() function
/external/tensorflow/tensorflow/core/lib/strings/
Dordered_code_test.cc1222 testing::BytesProcessed(n * len); in BM_WriteString()
1242 testing::BytesProcessed(n * len); in BM_ReadString()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc918 tensorflow::testing::BytesProcessed(static_cast<int64>(num_iters) * in BM_ParallelFusion()

12