Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/kernels/perception/
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
Dmax_unpooling_2d.cc34 const float* input_data, const int32_t* indices_data, in MaxUnpooling()
Dmax_pool_with_argmax.cc38 T* output_data, int32_t* indices_data) { in MaxPool()
/external/pytorch/aten/src/ATen/native/
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
DTensorDimApply.h15 T2* indices_data = indices.data_ptr<T2>(); in tensor_dim_apply3() local
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
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
DEmbedding.cpp138 auto indices_data = indices_contig.const_data_ptr<index_t>(); in embedding_dense_backward_cpu() local
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
/external/pytorch/aten/src/ATen/native/cuda/
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
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
DUnique.cu137 int64_t *indices_data = indices.mutable_data_ptr<int64_t>(); in unique_dim_cuda_template() local
DTensorModeKernel.cu163 int64_t* indices_data = indices.mutable_data_ptr<int64_t>(); in calculate_mode() local
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
/external/pytorch/aten/src/ATen/native/cpu/
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
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
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
DTensorCompareKernel.cpp258 int64_t* indices_data = (int64_t*)indices_data_bytes; in mode_kernel_impl() local
/external/tensorflow/tensorflow/lite/kernels/
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
Dembedding_lookup_sparse_test.cc57 std::initializer_list<int> indices_data, in SetInput()
Dsparse_to_dense.cc93 const auto indices_data = GetTensorData<T>(indices); in GetIndicesVector() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/
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
/external/executorch/kernels/portable/cpu/
Dop_topk.cpp65 long* indices_data = indices.mutable_data_ptr<long>(); in perform_topk() local
/external/tensorflow/tensorflow/lite/kernels/ctc/
Dctc_beam_search_decoder.cc157 int32_t* indices_data = GetTensorData<int32_t>(indices); in StoreAllDecodedSequences() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h606 const IndicesT* indices_data, in GatherNd()
634 const IndicesT* indices_data, in GatherNdString()
660 const IndicesT* indices_data, in ScatterNd()

12