OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:backward_conv_dnums
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc
189
ConvolutionDimensionNumbers
backward_conv_dnums
;
local
190
backward_conv_dnums
.set_input_batch_dimension(input_feature_dim);
191
backward_conv_dnums
.set_input_feature_dimension(input_batch_dim);
193
backward_conv_dnums
.add_input_spatial_dimensions(input_spatial_dims[i]);
195
backward_conv_dnums
.set_output_batch_dimension(kernel_input_feature_dim);
196
backward_conv_dnums
.set_output_feature_dimension(kernel_output_feature_dim);
198
backward_conv_dnums
.add_output_spatial_dimensions(kernel_spatial_dims[i]);
205
backward_conv_dnums
.set_kernel_input_feature_dimension(output_batch_dim);
206
backward_conv_dnums
.set_kernel_output_feature_dimension(output_feature_dim);
208
backward_conv_dnums
.add_kernel_spatial_dimensions(output_spatial_dims[i])
[
all
...]
pad_insertion.cc
221
ConvolutionDimensionNumbers
backward_conv_dnums
=
local
234
int64 dim =
backward_conv_dnums
.input_spatial_dimensions(i);
266
backward_conv_dnums
);
284
ConvolutionDimensionNumbers
backward_conv_dnums
=
local
318
int64 dim =
backward_conv_dnums
.output_spatial_dimensions(i);
331
backward_conv_dnums
);
355
int64 dim =
backward_conv_dnums
.output_spatial_dimensions(i);
Completed in 216 milliseconds