Searched refs:temp_indices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_topk.cc | 35 std::vector<tensorflow::int32> temp_indices(input_size); in TopK() local
37 std::iota(temp_indices.begin(), temp_indices.end(), 0); in TopK()
49 auto kth_element = temp_indices.begin() + k; in TopK()
50 std::partial_sort(temp_indices.begin(), kth_element, temp_indices.end(), in TopK()
64 std::copy(temp_indices.begin(), kth_element, out_indices_batch); in TopK()
66 out_values_batch[i] = values_batch[temp_indices[i]]; in TopK()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op_gpu.h | 474 Tensor temp_indices;
484 DT_INT32, TensorShape({num_rows, num_cols}), &temp_indices));
488 sorted_indices_ptr = temp_indices.flat<int32>().data();
544 To32Bit(temp_indices.matrix<int32>()).slice(slice_indices, slice_sizes);
|
| /external/freetype/src/sfnt/ |
| D | sfwoff2.c | 1763 WOFF2_Table* temp_indices = NULL; in woff2_open_font() local
2122 if ( FT_NEW_ARRAY( temp_indices, in woff2_open_font()
2128 temp_indices[nn] = indices[ttc_font->table_indices[nn]]; in woff2_open_font()
2137 indices[nn] = temp_indices[nn]; in woff2_open_font()
2139 FT_FREE( temp_indices ); in woff2_open_font()
|