HomeSort by relevance Sort by last modified time
    Searched refs:col_buffer_data (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
conv_grad_input_ops.cc 477 T* col_buffer_data = col_buffer.template flat<T>().data(); variable
500 TensorMap C(col_buffer_data, output_image_size, filter_total_size);
509 col_buffer_data, dims.in_depth, dims.spatial_dims[0].input_size,
533 &input_backprop_data, &col_buffer_data,
537 T* im2col_buf = col_buffer_data + shard_id * size_C;
    [all...]
conv_grad_filter_ops.cc 436 T* col_buffer_data = col_buffer.template flat<T>().data(); variable
462 auto shard = [&input_data, &col_buffer_data, &dims, &pad_top, &pad_left,
467 T* col_data_shard = col_buffer_data + shard_id * size_A;
482 ConstTensorMap A(col_buffer_data, output_image_size * shard_limit,
    [all...]

Completed in 118 milliseconds