| /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/ |
| fft.h | 143 bool in_place_fft, int batch_count) = 0;
166 bool in_place_fft, int batch_count,
237 bool in_place_fft, int batch_count) override; \
242 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) \
|
| stream.cc | [all...] |
| blas.h | [all...] |
| stream.h | [all...] |
| /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_fft.h | 72 int batch_count, ScratchAllocator* scratch_allocator);
|
| cuda_blas.h | 116 int batch_count, ScratchAllocator *scratch_allocator);
|
| cuda_blas.cc | [all...] |
| cuda_dnn.cc | 3620 const int batch_count = output_dimensions.NodesPerFeatureMap(); local
[all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_fft.h | 71 int batch_count, ScratchAllocator *scratch_allocator);
|
| rocm_fft.cc | 149 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) {
166 if (batch_count == 1 && input_embed == nullptr && output_embed == nullptr) {
286 output_distance, ROCMFftType(type), batch_count);
312 output_distance, ROCMFftType(type), batch_count, &size_in_bytes);
466 bool in_place_fft, int batch_count) {
471 batch_count, /*scratch_allocator=*/nullptr);
484 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) {
489 batch_count, scratch_allocator);
|
| rocm_blas.h | 109 int batch_count, ScratchAllocator *scratch_allocator);
|
| rocm_blas.cc | [all...] |
| rocm_dnn.cc | 3380 const int batch_count = output_dimensions.NodesPerFeatureMap(); local
[all...] |
| /external/tensorflow/tensorflow/python/keras/engine/ |
| training_dataset_test.py | 43 self.batch_count = 0
46 self.batch_count += 1
432 self.assertEqual(batch_counter.batch_count, 20)
455 self.assertEqual(batch_counter.batch_count, 20)
503 self.assertEqual(batch_counter.batch_count, 20)
541 self.assertEqual(batch_counter.batch_count, 10)
|
| training_utils.py | 606 batch_count = int(len(index_array) / batch_size)
609 last_batch = index_array[batch_count * batch_size:]
610 index_array = index_array[:batch_count * batch_size]
611 index_array = index_array.reshape((batch_count, batch_size))
[all...] |
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| metric_ops.py | [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/compiler/xla/service/cpu/ |
| dot_op_emitter.cc | 953 int64 batch_count = lhs_array_reshaped.GetShape().dimensions(0); 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...] |
