| /external/tensorflow/tensorflow/core/kernels/ |
| depthwise_conv_ops_test.cc | 57 const int batch_count = 1; local
62 Tensor image(dtype, {batch_count, image_height, image_width, depth});
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_fft.cc | 80 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) {
98 if (batch_count == 1 && input_embed == nullptr && output_embed == nullptr) {
196 output_stride, output_distance, CUDAFftType(type), batch_count);
220 output_stride, output_distance, CUDAFftType(type), batch_count,
393 bool in_place_fft, int batch_count) {
398 batch_count, /*scratch_allocator=*/nullptr);
408 << " batch_count: " << batch_count; local
420 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) {
425 batch_count, scratch_allocator)
435 << " batch_count: " << batch_count; local
[all...] |
| cuda_dnn.cc | 3620 const int batch_count = output_dimensions.NodesPerFeatureMap(); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| dot_op_emitter.cc | 953 int64 batch_count = lhs_array_reshaped.GetShape().dimensions(0); local
[all...] |
| /external/blktrace/iowatcher/ |
| main.c | 1104 int batch_count; local
1115 batch_count = cols / total_frames;
1117 if (batch_count == 0)
1118 batch_count = 1;
1168 while (i < cols && batch_i < batch_count) {
[all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_dnn.cc | 3380 const int batch_count = output_dimensions.NodesPerFeatureMap(); local
[all...] |
| /device/google/marlin/camera/QCamera2/HAL/ |
| QCameraParameters.cpp | 1282 int8_t batch_count = 0; local
7580 int32_t batch_count = m_captureFrameConfig.num_batch; local
7587 int32_t batch_count = m_captureFrameConfig.num_batch; local
[all...] |
