Home
last modified time | relevance | path

Searched refs:size_C (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc411 const size_t size_C = output_image_size * filter_total_size; in Compute() local
413 const size_t work_unit_size = size_A + size_B + size_C; in Compute()
507 &output_offset, &size_C](int64 start, int64 limit) { in Compute()
509 T* im2col_buf = col_buffer_data + shard_id * size_C; in Compute()
Dconv_grad_ops_3d.cc395 const int64 size_C = output_image_size * filter_total_size; in Compute() local
397 const int64 work_unit_size = size_A + size_B + size_C; in Compute()
530 &output_offset, &size_C](int64 start, int64 limit) { in Compute()
532 T* im2col_buf = col_buffer_data + shard_id * size_C; in Compute()
868 const int64 size_C = filter_total_size * dims.out_depth; in Compute() local
870 const int64 work_unit_size = size_A + size_B + size_C; in Compute()
Dconv_grad_filter_ops.cc318 const size_t size_C = filter_total_size * dims.out_depth; in Compute() local
320 const size_t work_unit_size = size_A + size_B + size_C; in Compute()