Home
last modified time | relevance | path

Searched refs:sum_j (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_Softmax.pbtxt19 $$softmax[i, j] = exp(logits[i, j]) / sum_j(exp(logits[i, j]))$$
Dapi_def_SegmentSum.pbtxt24 \\(output_i = \sum_j data_j\\) where sum is over `j` such
Dapi_def_SegmentMean.pbtxt24 \\(output_i = \frac{\sum_j data_j}{N}\\) where `mean` is
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3956 int sum_j = 0; in SoftmaxInt8LUT() local
3962 for (; sum_j <= last_dim - 16; sum_j += 16) { in SoftmaxInt8LUT()
3963 uint8x16_t input_value = vld1q_u8(input_data_uint + sum_j); in SoftmaxInt8LUT()
3986 for (; sum_j < last_dim; ++sum_j) { in SoftmaxInt8LUT()
3987 const uint8_t index = (input_data_uint[sum_j] ^ offset) + table_offset; in SoftmaxInt8LUT()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td13017 \\(output_i = \frac{\sum_j data_j}{N}\\) where `mean` is
13147 \\(output_i = \sum_j data_j\\) where sum is over `j` such
13746 $$softmax[i, j] = exp(logits[i, j]) / sum_j(exp(logits[i, j]))$$