Home
last modified time | relevance | path

Searched refs:batch_count (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.cc80 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize() argument
98 if (batch_count == 1 && input_embed == nullptr && output_embed == nullptr) { in Initialize()
196 output_stride, output_distance, CUDAFftType(type), batch_count); in Initialize()
220 output_stride, output_distance, CUDAFftType(type), batch_count, in Initialize()
393 bool in_place_fft, int batch_count) { in CreateBatchedPlan() argument
398 batch_count, /*scratch_allocator=*/nullptr); in CreateBatchedPlan()
408 << " batch_count: " << batch_count; in CreateBatchedPlan()
420 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator() argument
425 batch_count, scratch_allocator); in CreateBatchedPlanWithScratchAllocator()
435 << " batch_count: " << batch_count; in CreateBatchedPlanWithScratchAllocator()
Dcuda_blas.cc2150 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal() argument
2152 for (int i = 0; i < batch_count; ++i) { in DoBlasGemmBatchedInternal()
2160 const size_t size = batch_count * sizeof(CUDA_T *); in DoBlasGemmBatchedInternal()
2192 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasGemmBatchedInternal()
2194 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasGemmBatchedInternal()
2196 stream->AllocateTemporaryArray<CUDA_T *>(batch_count)); in DoBlasGemmBatchedInternal()
2234 batch_count, compute_type, algo); in DoBlasGemmBatchedInternal()
2249 const_cast<CUDA_T **>(GpuMemory(c)), ldc, batch_count); in DoBlasGemmBatchedInternal()
2257 for (int b = 0; b < batch_count; ++b) { in DoBlasGemmBatchedInternal()
2278 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched() argument
[all …]
Dcuda_fft.h72 int batch_count, ScratchAllocator* scratch_allocator);
Dcuda_blas.h116 int batch_count, ScratchAllocator *scratch_allocator);
Dcuda_dnn.cc3620 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
3622 std::vector<DeviceMemory<float>> a(batch_count); in DoMatMul()
3623 std::vector<DeviceMemory<float>> b(batch_count); in DoMatMul()
3624 std::vector<DeviceMemory<float>> c(batch_count); in DoMatMul()
3625 for (int i = 0; i < batch_count; ++i) { in DoMatMul()
3654 ldc, batch_count); in DoMatMul()
/external/tensorflow/tensorflow/stream_executor/
Dfft.h143 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) \
Dstream.cc4383 int batch_count) { in ThenBlasGemmBatched() argument
4385 b, ldb, beta, c, ldc, batch_count, in ThenBlasGemmBatched()
4395 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch() argument
4398 PARAM(beta), PARAM(c), PARAM(ldc), PARAM(batch_count)); in ThenBlasGemmBatchedWithScratch()
4407 k, alpha, a, lda, b, ldb, beta, c, ldc, batch_count, in ThenBlasGemmBatchedWithScratch()
4416 int batch_count) { in ThenBlasGemmBatched() argument
4418 b, ldb, beta, c, ldc, batch_count, in ThenBlasGemmBatched()
4427 int batch_count, ScratchAllocator *scratch_allocator) { in ThenBlasGemmBatchedWithScratch() argument
4430 PARAM(beta), PARAM(c), PARAM(ldc), PARAM(batch_count)); in ThenBlasGemmBatchedWithScratch()
4439 k, alpha, a, lda, b, ldb, beta, c, ldc, batch_count, in ThenBlasGemmBatchedWithScratch()
[all …]
Dblas.h1093 int ldc, int batch_count, ScratchAllocator *scratch_allocator) = 0;
1100 int batch_count, ScratchAllocator *scratch_allocator) = 0;
1107 int batch_count, ScratchAllocator *scratch_allocator) = 0;
1115 int batch_count, ScratchAllocator *scratch_allocator) = 0;
1123 int batch_count, ScratchAllocator *scratch_allocator) = 0;
1131 int64 stride_c, int batch_count) = 0;
1137 int batch_count) = 0;
1143 int batch_count) = 0;
1150 int64 stride_c, int batch_count) = 0;
1157 int64 stride_c, int batch_count) = 0;
[all …]
Dstream.h1382 int ldc, int batch_count);
1390 int ldc, int batch_count);
1398 int ldc, int batch_count);
1406 int batch_count);
1414 int batch_count);
1421 int ldc, int batch_count, ScratchAllocator *scratch_allocator);
1427 int batch_count, ScratchAllocator *scratch_allocator);
1433 int batch_count, ScratchAllocator *scratch_allocator);
1441 int batch_count, ScratchAllocator *scratch_allocator);
1449 int batch_count, ScratchAllocator *scratch_allocator);
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.cc149 fft::Type type, int batch_count, ScratchAllocator *scratch_allocator) { in Initialize() argument
166 if (batch_count == 1 && input_embed == nullptr && output_embed == nullptr) { in Initialize()
286 output_distance, ROCMFftType(type), batch_count); in Initialize()
312 output_distance, ROCMFftType(type), batch_count, &size_in_bytes); in Initialize()
466 bool in_place_fft, int batch_count) { in CreateBatchedPlan() argument
471 batch_count, /*scratch_allocator=*/nullptr); in CreateBatchedPlan()
484 bool in_place_fft, int batch_count, ScratchAllocator *scratch_allocator) { in CreateBatchedPlanWithScratchAllocator() argument
489 batch_count, scratch_allocator); in CreateBatchedPlanWithScratchAllocator()
Drocm_blas.cc1814 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatchedInternal() argument
1821 for (int i = 0; i < batch_count; ++i) { in DoBlasGemmBatchedInternal()
1841 if (batch_count > 1) { in DoBlasGemmBatchedInternal()
1852 for (int i = 1; i < batch_count - 1; ++i) { in DoBlasGemmBatchedInternal()
1893 GpuComplex(beta_ptr), c_raw_ptrs[0], ldc, bsc, batch_count); in DoBlasGemmBatchedInternal()
1910 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched() argument
1916 alpha_half, a, lda, b, ldb, beta_half, c, ldc, batch_count, in DoBlasGemmBatched()
1931 int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched() argument
1934 alpha, a_array, lda, b_array, ldb, beta, c_array, ldc, batch_count, in DoBlasGemmBatched()
1948 int ldc, int batch_count, ScratchAllocator *scratch_allocator) { in DoBlasGemmBatched() argument
[all …]
Drocm_fft.h71 int batch_count, ScratchAllocator *scratch_allocator);
Drocm_blas.h109 int batch_count, ScratchAllocator *scratch_allocator);
Drocm_dnn.cc3380 const int batch_count = output_dimensions.NodesPerFeatureMap(); in DoMatMul() local
3382 std::vector<DeviceMemory<float>> a(batch_count); in DoMatMul()
3383 std::vector<DeviceMemory<float>> b(batch_count); in DoMatMul()
3384 std::vector<DeviceMemory<float>> c(batch_count); in DoMatMul()
3385 for (int i = 0; i < batch_count; ++i) { in DoMatMul()
3414 ldc, batch_count); in DoMatMul()
/external/tensorflow/tensorflow/core/kernels/
Ddepthwise_conv_ops_test.cc57 const int batch_count = 1; in Run() local
62 Tensor image(dtype, {batch_count, image_height, image_width, depth}); in Run()
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_dataset_test.py43 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)
Dtraining_utils.py606 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))
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dmetric_ops.py3208 batch_count = math_ops.cast(
3214 batch_count = math_ops.reduce_sum(weights) # n_B in eqn
3218 update_count = state_ops.assign_add(count_, batch_count) # n_AB in eqn
3219 prev_count = update_count - batch_count # n_A in update equation
3224 math_ops.reduce_sum(weighted_predictions), batch_count)
3226 (batch_mean_prediction - mean_prediction) * batch_count, update_count)
3234 math_ops.reduce_sum(weighted_labels), batch_count)
3236 (batch_mean_label - mean_label) * batch_count, update_count)
3256 (prev_count * batch_count / update_count))
/external/blktrace/iowatcher/
Dmain.c1104 int batch_count; in plot_io_movie() local
1115 batch_count = cols / total_frames; in plot_io_movie()
1117 if (batch_count == 0) in plot_io_movie()
1118 batch_count = 1; in plot_io_movie()
1168 while (i < cols && batch_i < batch_count) { in plot_io_movie()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Ddot_op_emitter.cc953 int64 batch_count = lhs_array_reshaped.GetShape().dimensions(0); in EmitBatchDotOperation() local
958 llvm_ir::IrName(&dot, "bdot"), /*start=*/0, /*end=*/batch_count, in EmitBatchDotOperation()