Home
last modified time | relevance | path

Searched refs:unique_index (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dunique.cc88 const int unique_index = unique_values.size(); in EvalImpl() local
89 unique_values[data[i]] = unique_index; in EvalImpl()
90 indexes[i] = unique_index; in EvalImpl()
/external/pytorch/aten/src/ATen/native/
DUnique.cpp179 Tensor unique_index = at::empty({0}, self.options().dtype(kLong)); in unique_cpu_sorted_template() local
229 unique_index.resize_({unique_count + 1}); in unique_cpu_sorted_template()
230 unique_index_data = unique_index.data_ptr<int64_t>(); in unique_cpu_sorted_template()
/external/libxaac/encoder/drc_src/
Dimpd_drc_mux.c794 WORD32 *unique_index; in impd_drc_write_drc_instruct_uni_drc() local
812 unique_index = (WORD32 *)ptr_scratch; in impd_drc_write_drc_instruct_uni_drc()
814 (FLOAT32 *)((UWORD8 *)ptr_scratch + (MAX_CHANNEL_COUNT) * sizeof(unique_index[0])); in impd_drc_write_drc_instruct_uni_drc()
935 unique_index[j] = -10; in impd_drc_write_drc_instruct_uni_drc()
949 if ((index >= 0) && (unique_index[n] == index) && (unique_scaling[n] == factor)) { in impd_drc_write_drc_instruct_uni_drc()
956 unique_index[g] = index; in impd_drc_write_drc_instruct_uni_drc()
970 if (((index >= 0) && (unique_index[n] == index)) || in impd_drc_write_drc_instruct_uni_drc()
978 unique_index[g] = index; in impd_drc_write_drc_instruct_uni_drc()
997 pstr_drc_instructions_uni_drc->gain_set_index_for_channel_group[g] = unique_index[g]; in impd_drc_write_drc_instruct_uni_drc()
1055 unique_index[i] = -1; in impd_drc_write_drc_instruct_uni_drc()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc508 static mlir::ElementsAttr GetSplat(RankedTensorType type, int unique_index, in GetSplat() argument
514 type, builder.getIntegerAttr(element_ty, unique_index)); in GetSplat()
518 type, builder.getFloatAttr(element_ty, unique_index)); in GetSplat()
524 new_type, builder.getIntegerAttr(qtype.getStorageType(), unique_index)); in GetSplat()