Home
last modified time | relevance | path

Searched defs:inner_index (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dlog_softmax.h45 for (int inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmax() local
52 for (int inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmax() local
80 for (int inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmax() local
Dl2normalization.h37 for (int inner_index = 0; inner_index < depth; ++inner_index) { in L2Normalization() local
47 for (int inner_index = 0; inner_index < depth; ++inner_index) { in L2Normalization() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlog_softmax.h181 for (size_t inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmaxQuantized() local
188 for (size_t inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmaxQuantized() local
217 for (size_t inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmaxQuantized() local
Dcumsum.h54 for (size_t inner_index = 0; inner_index < inner; inner_index++) { in CumSum() local
122 for (size_t inner_index = 0; inner_index < inner; inner_index++) { in CumSum() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_scatter_utils.cc60 int64_t inner_index = GetIndexForId(index->operand(1), id, map_id); in GetIndexForId() local