OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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