OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
defs:output_rank
(Results
1 - 3
of
3
) sorted by null
/external/tensorflow/tensorflow/compiler/xla/service/
instruction_fusion.cc
135
int64
output_rank
= 0;
local
138
[&
output_rank
](const Shape& subshape, const ShapeIndex& shape_index) {
140
output_rank
= std::max(
output_rank
, ShapeUtil::TrueRank(subshape));
144
[
output_rank
](HloInstruction* operand) {
153
output_rank
;
/external/tensorflow/tensorflow/core/kernels/
reshape_util.cc
51
const int64
output_rank
= target_shape_in.NumElements();
local
62
for (int d = 0; d <
output_rank
; ++d) {
115
gtl::InlinedVector<int64, 8> output_strides(
output_rank
);
116
output_strides[
output_rank
- 1] = 1;
117
for (int d =
output_rank
- 2; d >= 0; --d) {
124
TensorShape({nnz,
output_rank
}),
133
for (int j = 0; j <
output_rank
; ++j) {
141
TensorShape({
output_rank
}),
/external/tensorflow/tensorflow/contrib/lite/kernels/
embedding_lookup_sparse.cc
154
const int
output_rank
= (lookup_rank - 1) + (embedding_rank - 1);
local
161
TfLiteIntArray* output_shape = TfLiteIntArrayCreate(
output_rank
);
Completed in 85 milliseconds