OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:before_dim
(Results
1 - 2
of
2
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
unpack_op.cc
93
int64
before_dim
= 1;
variable
95
before_dim
*= input_shape.dim_size(i);
107
input.shaped<T, 3>({1,
before_dim
, axis_dim * after_dim});
115
auto output_shaped = output->shaped<T, 3>({1,
before_dim
, after_dim});
117
Eigen::DSizes<Eigen::DenseIndex, 3> sizes{1,
before_dim
, after_dim};
pack_op.cc
90
int64
before_dim
= 1;
variable
92
before_dim
*= output_shape.dim_size(i);
105
output->shaped<T, 2>({
before_dim
, after_dim * axis_dim});
113
values[i].shaped<T, 2>({
before_dim
, after_dim})));
Completed in 83 milliseconds