Home
last modified time | relevance | path

Searched defs:indices_data (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/perception/
H A Dmax_unpooling_2d_test.cc102 std::vector<int32_t> indices_data{0, 3, 4, 7, 8, 11, 12, 15}; in TEST() local
127 std::vector<int32_t> indices_data{0, 5, 10, 13, 19, 20, 27, 31}; in TEST() local
154 std::vector<int32_t> indices_data{6, 9, 16, 19}; in TEST() local
182 std::vector<int32_t> indices_data{2, 23, 8, 9, 12, 15, 40, 43, 44, 47, 72, in TEST() local
223 std::vector<int32_t> indices_data{2, 23, 8, 9, 12, 15, 40, 43, 44, 47, 72, in TEST() local
H A Dmax_unpooling_2d.cc34 const float* input_data, const int32_t* indices_data, in MaxUnpooling()
H A Dmax_pool_with_argmax.cc38 T* output_data, int32_t* indices_data) { in MaxPool()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DAdaptiveMaxPooling3d.cpp164 int64_t *indices_data, in adaptive_max_pool3d_out_frame()
235 const int64_t *indices_data, in adaptive_max_pool3d_backward_out_frame()
304 auto indices_data = indices.data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
327 auto indices_data = indices.data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
398 const int64_t* indices_data = indices.const_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
418 const int64_t* indices_data = indices.const_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
H A DEmbeddingBag.cpp1077 auto* indices_data = indices.const_data_ptr<index_t>(); in embedding_bag_cpu_max_out() local
1466 const index_t* indices_data, in compute_counts()
1487 const index_t* indices_data, in compute_counts_uniq()
1542 auto* indices_data = indices.const_data_ptr<index_t>(); in _embedding_bag_dense_backward_cpu_sum_mean() local
1704 auto* indices_data = indices.const_data_ptr<index_t>(); in _embedding_bag_per_sample_weights_backward_cpu_template() local
H A DFractionalMaxPool2d.cpp326 auto indices_data = indices.data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
382 auto indices_data = indices.const_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
H A DUnique.cpp188 int64_t* indices_data = indices.data_ptr<int64_t>(); in unique_cpu_sorted_template() local
336 int64_t *indices_data = indices.data(); in _unique_dim_cpu_impl() local
H A DTensorDimApply.h15 T2* indices_data = indices.data_ptr<T2>(); in tensor_dim_apply3() local
H A DEmbedding.cpp138 auto indices_data = indices_contig.const_data_ptr<index_t>(); in embedding_dense_backward_cpu() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DAdaptiveMaxPooling3d.cu126 int64_t *indices_data, in adaptivemaxpool_loop()
206 const int64_t *indices_data, in adaptivemaxgradinput_loop()
282 const int64_t *indices_data, in atomicadaptivemaxgradinput_loop()
365 int64_t* indices_data = indices.mutable_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
449 const int64_t* indices_data = indices.const_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
472 const int64_t* indices_data = indices.const_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
H A DAdaptiveMaxPooling2d.cu238 int64_t* indices_data = indices_c.mutable_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
287 int64_t* indices_data = indices_c.mutable_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
371 const int64_t* indices_data = indices_.const_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
432 const int64_t* indices_data = indices_.const_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
H A DDilatedMaxPool2d.cu353 int64_t *indices_data = indices.mutable_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
489 const int64_t *indices_data = indices.const_data_ptr<int64_t>(); in TORCH_IMPL_FUNC() local
H A DUnique.cu137 int64_t *indices_data = indices.mutable_data_ptr<int64_t>(); in unique_dim_cuda_template() local
H A DTensorModeKernel.cu163 int64_t* indices_data = indices.mutable_data_ptr<int64_t>(); in calculate_mode() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAdaptiveMaxPoolKernel.cpp29 auto indices_data = indices.data_ptr<int64_t>(); in cpu_adaptive_max_pool2d() local
100 auto indices_data = indices.data_ptr<int64_t>(); in cpu_adaptive_max_pool2d_channels_last() local
217 auto indices_data = indices.data_ptr<int64_t>(); in cpu_adaptive_max_pool2d_channels_last() local
351 auto indices_data = indices.const_data_ptr<int64_t>(); in cpu_adaptive_max_pool2d_backward() local
401 auto indices_data = indices.const_data_ptr<int64_t>(); in cpu_adaptive_max_pool2d_backward_channels_last() local
495 auto indices_data = indices.data_ptr<int64_t>(); in cpu_adaptive_max_pool3d() local
574 auto indices_data = indices.data_ptr<int64_t>(); in cpu_adaptive_max_pool3d_channels_last() local
699 auto indices_data = indices.data_ptr<int64_t>(); in cpu_adaptive_max_pool3d_channels_last() local
841 auto indices_data = indices.data_ptr<int64_t>(); in cpu_adaptive_max_pool3d_backward() local
895 auto indices_data = indices.data_ptr<int64_t>(); in cpu_adaptive_max_pool3d_backward_channels_last() local
H A DMaxUnpoolKernel.cpp24 auto indices_data = indices.const_data_ptr<int64_t>(); in cpu_max_unpool() local
109 auto indices_data = indices.const_data_ptr<int64_t>(); in cpu_max_unpool_channels_last() local
169 auto indices_data = indices.data_ptr<int64_t>(); in cpu_max_unpool_backward() local
H A DMaxPoolKernel.cpp267 auto indices_data = indices.data_ptr<int64_t>(); in cpu_max_pool() local
397 auto indices_data = indices.data_ptr<int64_t>(); in cpu_max_pool_channels_last() local
482 auto indices_data = indices.const_data_ptr<int64_t>(); in cpu_max_pool_backward() local
556 auto indices_data = indices.const_data_ptr<int64_t>(); in cpu_max_pool_backward_channels_last() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Ddynamic_update_slice.cc95 std::vector<int> ClampStartIndices(int input_dims, const int32_t* indices_data, in ClampStartIndices()
113 const int32_t* indices_data = GetTensorData<int32_t>(indice); in DynamicUpdateSlice() local
H A Dembedding_lookup_sparse_test.cc57 std::initializer_list<int> indices_data, in SetInput()
H A Dsparse_to_dense.cc93 const auto indices_data = GetTensorData<T>(indices); in GetIndicesVector() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag.cpp50 const auto indices_data = indices.data_ptr<IndexType>(); in embedding_lookup_fallback_impl() local
297 const auto indices_data = indices.data_ptr<IndexType>(); in embedding_bag_nbit_impl() local
469 const auto indices_data = indices.data_ptr<IndexType>(); in embedding_bag_byte_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dreference_ops.h606 const IndicesT* indices_data, in GatherNd()
634 const IndicesT* indices_data, in GatherNdString()
660 const IndicesT* indices_data, in ScatterNd()
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_topk.cpp65 long* indices_data = indices.mutable_data_ptr<long>(); in perform_topk() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ctc/
H A Dctc_beam_search_decoder.cc157 int32_t* indices_data = GetTensorData<int32_t>(indices); in StoreAllDecodedSequences() local
/aosp_15_r20/external/pytorch/torch/sparse/
H A D_triton_ops.py260 def scatter_mm(blocks, others, indices_data, *, accumulators=None): argument
1022 def bsr_scatter_mm(bsr, other, indices_data=None, out=None): argument

12