OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:transpose_order
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
depthtospace_op.cc
60
std::vector<int64>
transpose_order
;
variable
63
transpose_order
.reserve(input_rank);
77
transpose_order
.push_back(0);
79
transpose_order
.push_back(i + 1);
80
transpose_order
.push_back(i + 1 + num_spatial_dims);
82
transpose_order
.push_back(feature_dim + num_spatial_dims);
102
transpose_order
.push_back(0);
103
transpose_order
.push_back(1 + num_spatial_dims);
105
transpose_order
.push_back(2 + num_spatial_dims + i);
106
transpose_order
.push_back(1 + i)
[
all
...]
spacetodepth_op.cc
60
std::vector<int64>
transpose_order
;
variable
63
transpose_order
.reserve(input_rank);
82
transpose_order
.push_back(0);
84
transpose_order
.push_back(i * 2 + 1);
87
transpose_order
.push_back(i * 2 + 2);
89
transpose_order
.push_back(feature_dim + num_spatial_dims);
114
transpose_order
.push_back(0);
116
transpose_order
.push_back(i * 2 + 3);
118
transpose_order
.push_back(feature_dim);
120
transpose_order
.push_back(i * 2 + 2)
[
all
...]
Completed in 178 milliseconds