OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:conv_dnums
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_rewriter.cc
79
const ConvolutionDimensionNumbers&
conv_dnums
=
local
81
auto input_batch_dim =
conv_dnums
.input_batch_dimension();
82
auto input_feature_dim =
conv_dnums
.input_feature_dimension();
83
auto input_spatial_dims =
conv_dnums
.input_spatial_dimensions();
84
auto kernel_input_feature_dim =
conv_dnums
.kernel_input_feature_dimension();
85
auto kernel_output_feature_dim =
conv_dnums
.kernel_output_feature_dimension();
86
auto kernel_spatial_dims =
conv_dnums
.kernel_spatial_dimensions();
87
auto output_batch_dim =
conv_dnums
.output_batch_dimension();
88
auto output_feature_dim =
conv_dnums
.output_feature_dimension();
89
auto output_spatial_dims =
conv_dnums
.output_spatial_dimensions()
[
all
...]
cudnn_convolution_rewriter_test.cc
256
ConvolutionDimensionNumbers
conv_dnums
;
local
257
conv_dnums
.set_input_batch_dimension(0);
258
conv_dnums
.set_output_batch_dimension(0);
259
conv_dnums
.set_input_feature_dimension(1);
260
conv_dnums
.set_output_feature_dimension(1);
261
conv_dnums
.add_input_spatial_dimensions(2);
262
conv_dnums
.add_output_spatial_dimensions(2);
263
conv_dnums
.add_input_spatial_dimensions(3);
264
conv_dnums
.add_output_spatial_dimensions(3);
265
conv_dnums
.set_kernel_input_feature_dimension(0)
[
all
...]
pad_insertion.cc
43
const Window& conv_window, const ConvolutionDimensionNumbers&
conv_dnums
,
58
for (size_t i = 0; i <
conv_dnums
.input_spatial_dimensions().size(); ++i) {
59
int64 dim =
conv_dnums
.input_spatial_dimensions(i);
90
for (size_t i = 0; i <
conv_dnums
.input_spatial_dimensions().size(); ++i) {
91
int64 dim =
conv_dnums
.input_spatial_dimensions(i);
115
const ConvolutionDimensionNumbers&
conv_dnums
,
126
for (size_t i = 0; i <
conv_dnums
.kernel_spatial_dimensions().size(); ++i) {
127
int64 dim =
conv_dnums
.kernel_spatial_dimensions(i);
Completed in 1786 milliseconds