OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:dimension_numbers
(Results
1 - 4
of
4
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/
reference_util_test.cc
333
ConvolutionDimensionNumbers
dimension_numbers
;
local
334
dimension_numbers
.set_input_batch_dimension(2);
335
dimension_numbers
.set_input_feature_dimension(0);
336
dimension_numbers
.set_output_batch_dimension(2);
337
dimension_numbers
.set_output_feature_dimension(0);
338
dimension_numbers
.add_input_spatial_dimensions(1);
339
dimension_numbers
.add_output_spatial_dimensions(1);
340
dimension_numbers
.add_input_spatial_dimensions(3);
341
dimension_numbers
.add_output_spatial_dimensions(3);
342
dimension_numbers
.set_kernel_output_feature_dimension(0)
389
ConvolutionDimensionNumbers
dimension_numbers
;
local
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment.cc
74
const ConvolutionDimensionNumbers&
dimension_numbers
=
local
77
for (int i =
dimension_numbers
.input_spatial_dimensions_size() - 1; i >= 0;
79
input_layout.push_back(
dimension_numbers
.input_spatial_dimensions(i));
81
input_layout.push_back(
dimension_numbers
.input_feature_dimension());
82
input_layout.push_back(
dimension_numbers
.input_batch_dimension());
86
for (int i =
dimension_numbers
.kernel_spatial_dimensions_size() - 1; i >= 0;
88
filter_layout.push_back(
dimension_numbers
.kernel_spatial_dimensions(i));
90
filter_layout.push_back(
dimension_numbers
.kernel_input_feature_dimension());
91
filter_layout.push_back(
dimension_numbers
.kernel_output_feature_dimension());
95
for (int i =
dimension_numbers
.output_spatial_dimensions_size() - 1; i >= 0
[
all
...]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
image_resize_ops.cc
156
xla::ConvolutionDimensionNumbers
dimension_numbers
;
local
157
dimension_numbers
.set_input_batch_dimension(0);
158
dimension_numbers
.set_output_batch_dimension(0);
159
dimension_numbers
.set_input_feature_dimension(3);
160
dimension_numbers
.set_output_feature_dimension(3);
162
dimension_numbers
.add_input_spatial_dimensions(1 + i);
163
dimension_numbers
.add_output_spatial_dimensions(1 + i);
164
dimension_numbers
.add_kernel_spatial_dimensions(i);
166
dimension_numbers
.set_kernel_input_feature_dimension(num_spatial_dims);
167
dimension_numbers
.set_kernel_output_feature_dimension(num_spatial_dims + 1)
201
xla::ConvolutionDimensionNumbers
dimension_numbers
;
local
[
all
...]
/external/tensorflow/tensorflow/compiler/xla/client/
computation_builder.cc
540
DotDimensionNumbers
dimension_numbers
;
local
541
dimension_numbers
.add_lhs_contracting_dimensions(
543
dimension_numbers
.add_rhs_contracting_dimensions(0);
544
return DotGeneral(lhs, rhs,
dimension_numbers
);
549
const DotDimensionNumbers&
dimension_numbers
) {
554
*request->mutable_dimension_numbers() =
dimension_numbers
;
576
const ConvolutionDimensionNumbers&
dimension_numbers
) {
618
dimension_numbers
.input_spatial_dimensions()) &&
621
dimension_numbers
.kernel_spatial_dimensions()) &&
624
dimension_numbers
.output_spatial_dimensions())
1514
ConvolutionDimensionNumbers
dimension_numbers
;
local
1564
ConvolutionDimensionNumbers
dimension_numbers
;
local
[
all
...]
Completed in 270 milliseconds