Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_filter_ops.cc314 const size_t size_A = output_image_size * filter_total_size; in Compute() local
320 const size_t work_unit_size = size_A + size_B + size_C; in Compute()
370 &size_A](int64 start, int64 limit) { in Compute()
373 T* col_data_shard = col_buffer_data + shard_id * size_A; in Compute()
386 size_A, shard); in Compute()
Dconv_grad_ops_3d.cc391 const int64 size_A = output_image_size * dims.out_depth; in Compute() local
397 const int64 work_unit_size = size_A + size_B + size_C; in Compute()
864 const int64 size_A = output_image_size * filter_total_size; in Compute() local
870 const int64 work_unit_size = size_A + size_B + size_C; in Compute()
950 &size_A](int64 start, int64 limit) { in Compute()
953 T* col_data_shard = col_buffer_data + shard_id * size_A; in Compute()
977 size_A, shard); in Compute()
Dconv_grad_input_ops.cc407 const size_t size_A = output_image_size * dims.out_depth; in Compute() local
413 const size_t work_unit_size = size_A + size_B + size_C; in Compute()