Searched refs:selected_index (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | quantized_ops_test.py | 72 selected_index = math_ops.equal(
74 gather_index = array_ops.boolean_mask(iota_r1, selected_index)
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | detection_postprocess.cc | 542 int selected_index = selected[i]; in NonMaxSuppressionMultiClassRegularHelper() local
545 (selected_index * num_classes_with_background + col + label_offset); in NonMaxSuppressionMultiClassRegularHelper()
547 class_scores[selected_index]; in NonMaxSuppressionMultiClassRegularHelper()
672 int selected_index = selected[i]; in NonMaxSuppressionMultiClassFastHelper() local
675 scores + selected_index * num_classes_with_background + label_offset; in NonMaxSuppressionMultiClassFastHelper()
677 sorted_class_indices + selected_index * num_classes; in NonMaxSuppressionMultiClassFastHelper()
686 reinterpret_cast<BoxCornerEncoding*>(decoded_boxes)[selected_index]; in NonMaxSuppressionMultiClassFastHelper()
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | dist_diversification.cc | 41 for (const int selected_index : result) { in DiversifyByDistance() local
42 if (distance_matrix(index, selected_index) < min_distance) { in DiversifyByDistance()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | detection_postprocess.cc | 570 for (const auto& selected_index : selected) { in NonMaxSuppressionMultiClassRegularHelper() local
572 (selected_index * num_classes_with_background + col + label_offset); in NonMaxSuppressionMultiClassRegularHelper()
574 class_scores[selected_index]; in NonMaxSuppressionMultiClassRegularHelper()
703 for (const auto& selected_index : selected) { in NonMaxSuppressionMultiClassFastHelper() local
705 scores + selected_index * num_classes_with_background + label_offset; in NonMaxSuppressionMultiClassFastHelper()
707 sorted_class_indices.data() + selected_index * num_classes; in NonMaxSuppressionMultiClassFastHelper()
716 decoded_boxes)[selected_index]; in NonMaxSuppressionMultiClassFastHelper()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | clustering_ops.cc | 254 int64 selected_index = 0; in Compute() local
255 float selected_distance = distances(selected_index); in Compute()
262 selected_index = i; in Compute()
273 output() = selected_index; in Compute()
|
| /external/mesa3d/src/vulkan/device-select-layer/ |
| D | device_select_layer.c | 449 …unsigned selected_index = get_default_device(info, selection, physical_device_count, physical_devi… in device_select_EnumeratePhysicalDevices() local
451 selected_physical_devices[0] = physical_devices[selected_index]; in device_select_EnumeratePhysicalDevices()
453 unsigned this_idx = i < selected_index ? i : i + 1; in device_select_EnumeratePhysicalDevices()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator_typed_visitor.h | 1893 absl::optional<std::vector<int64>> selected_index;
1901 selected_index = operand_index;
1913 selected_index = operand_index;
1922 selected_index->begin())) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 821 llvm_ir::IrArray::Index selected_index( in HandleSelectAndScatter() local
824 output_array.EmitReadArrayElement(selected_index, &b_); in HandleSelectAndScatter()
828 output_array.EmitWriteArrayElement(selected_index, scatter_value, &b_); in HandleSelectAndScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2403 IrArray::Index selected_index(selected_multi_index, output_shape, in EmitSelectAndScatterFromMlir() local
2406 out_array.EmitArrayElementAddress(selected_index, &b_); in EmitSelectAndScatterFromMlir()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 951 Value selected_index = builder->create<SelectOp>( in BuildArgMinMaxReductionBody() local
954 Value return_values[] = {selected_input, selected_index}; in BuildArgMinMaxReductionBody()
|