Home
last modified time | relevance | path

Searched defs:output_indices (Results 1 – 25 of 44) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dnon_max_suppression_op.cu.cc328 Tensor* output_indices = nullptr; in DoNMS() local
402 Tensor* output_indices = nullptr; in DoNMS() local
427 Tensor* output_indices = nullptr; in DoNMS() local
546 Tensor* output_indices = nullptr; in Compute() local
592 Tensor* output_indices = nullptr; in Compute() local
643 Tensor* output_indices = nullptr; in Compute() local
/aosp_15_r20/external/libtextclassifier/native/utils/tflite/
H A Ddist_diversification.cc83 TfLiteTensor& output_indices = in AllocateOutputIndexes() local
97 TfLiteTensor& output_indices = in Prepare() local
110 TfLiteTensor& output_indices = in Eval() local
/aosp_15_r20/external/ComputeLibrary/src/core/CPP/kernels/
H A DCPPNonMaximumSuppressionKernel.cpp38 …ments(const ITensorInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *output_indices, uns… in validate_arguments()
62 …ernel::configure(const ITensor *input_bboxes, const ITensor *input_scores, ITensor *output_indices, in configure()
85 …::validate(const ITensorInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *output_indices, in validate()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddeserialize_sparse_string_op.cc82 Tensor output_indices; in Compute() local
105 Tensor output_indices; in Compute() local
232 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor()
H A Ddeserialize_sparse_variant_op.cc59 const Tensor* output_indices; in Compute() local
152 Tensor* output_indices; in Compute() local
304 int index, int expected_rank, const Tensor** output_indices, in GetAndValidateSparseTensorIndicesAndValues()
H A Dsparse_slice_op_gpu.cu.cc94 int64_t* __restrict__ output_indices, T* __restrict__ output_values) { in SparseSliceGatherKernel()
152 Tensor* output_indices = nullptr; in operator ()() local
216 Tensor* output_indices = nullptr; in operator ()() local
H A Dsparse_fill_empty_rows_op_gpu.cu.cc132 const Tindex* num_new_rows_before, Tindex* output_indices, T* output_values, in ScatterInputElementsKernel()
154 const bool* empty_row_indicator, Tindex* output_indices, T* output_values) { in ScatterNewElementsKernel()
190 Tindex* output_indices; in operator ()() local
374 Tindex* output_indices; in operator ()() local
H A Dserialize_sparse_op.cc195 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in operator ()() local
296 Tensor& output_indices = serialized_sparse_t(b, 0).emplace<Tensor>( in operator ()() local
H A Dsparse_split_op_gpu.cu.cc168 Index* __restrict__* __restrict__ output_indices = in SparseSplitScatterKernel() local
313 GpuDeviceArrayOnHost<Index*> output_indices(context, num_split); in operator ()() local
H A Dsparse_tensors_map_ops.cc307 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in Compute() local
399 const Tensor& output_indices = st.indices(); in Compute() local
H A Dsparse_slice_grad_op.cc80 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
H A Dreshape_util_gpu.cu.cc35 Tindex* __restrict__ output_indices) { in ReshapeSparseTensorKernel()
H A Dsparse_slice_grad_op_gpu.cu.cc97 MultiIndexSearchIterator output_indices, in SparseSliceGradKernel()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dsparse_ops.cc331 ShapeHandle output_indices = in __anone91b66370e02() local
374 ShapeHandle output_indices = in __anone91b66371002() local
628 ShapeHandle output_indices = in __anone91b66371802() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dquantization_util.cc108 TfLiteContext* context, const std::vector<uint32_t>& output_indices, in QuantizeOutputs()
118 TfLiteContext* context, const std::vector<int64_t>& output_indices, in QuantizeOutputs()
H A Dquantization_util_test.cc211 std::vector<uint32_t> output_indices = {0}; in TEST() local
233 std::vector<int64_t> output_indices = {0}; in TEST() local
H A Doperations.h178 bool output_indices = false; member
190 bool output_indices = false; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dpooling.cc130 bool output_indices, GPUOperation* op) { in GetMaxPoolingKernelCode()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreverse_test.cc93 std::vector<int64_t> output_indices(spec.input_dims.size()); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_binary.cc127 const std::vector<int> output_indices = ReverseOffset(output_shape, k); in EvaluateBinaryOperatorOnConstantInputs() local
H A Dresolve_constant_unary.cc66 std::vector<int> output_indices(input_shape.dimensions_count()); in ReduceGeneric() local
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/
H A Dmultiplex_3_kernel.cc152 auto output_indices = output_indices_tensor->matrix<int64_t>(); in Compute() local
/aosp_15_r20/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1/src/main/proto/google/cloud/aiplatform/v1/
H A Dexplanation.proto264 google.protobuf.ListValue output_indices = 5; field
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseMatMul.cpp236 auto output_indices = output._indices(); in sparse_matmul_kernel() local
/aosp_15_r20/external/google-cloud-java/java-aiplatform/proto-google-cloud-aiplatform-v1beta1/src/main/proto/google/cloud/aiplatform/v1beta1/
H A Dexplanation.proto270 google.protobuf.ListValue output_indices = 5; field

12