| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | AdaptiveMaxPoolKernel.cpp | 20 const Tensor& indices_, in cpu_adaptive_max_pool2d()
88 const Tensor& indices_, in cpu_adaptive_max_pool2d_channels_last()
205 const Tensor& indices_, in cpu_adaptive_max_pool2d_channels_last()
345 const Tensor& indices_) { in cpu_adaptive_max_pool2d_backward()
391 const Tensor& indices_) { in cpu_adaptive_max_pool2d_backward_channels_last()
486 const Tensor& indices_, in cpu_adaptive_max_pool3d()
562 const Tensor& indices_, in cpu_adaptive_max_pool3d_channels_last()
687 const Tensor& indices_, in cpu_adaptive_max_pool3d_channels_last()
835 const Tensor& indices_) { in cpu_adaptive_max_pool3d_backward()
885 const Tensor& indices_) { in cpu_adaptive_max_pool3d_backward_channels_last()
|
| H A D | MaxPoolKernel.cpp | 238 const Tensor& indices_, in cpu_max_pool()
358 const Tensor& indices_, in cpu_max_pool_channels_last()
476 const Tensor& indices_) { in cpu_max_pool_backward()
541 const Tensor& indices_) { in cpu_max_pool_backward_channels_last()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | MaxUnpooling.cu | 117 const Tensor& indices_, in max_unpooling2d_forward_out_cuda()
292 const Tensor& indices_, in max_unpooling3d_forward_out_cuda()
402 const Tensor& indices_, in max_unpooling2d_backward_out_cuda()
502 const Tensor& indices_, in max_unpooling3d_backward_out_cuda()
|
| H A D | ScanKernels.cpp | 35 auto indices_ = contiguous_out_arg(indices); in cummax_helper_cuda() local
52 auto indices_ = contiguous_out_arg(indices); in cummin_helper_cuda() local
|
| H A D | EmbeddingBag.cu | 170 const Tensor &indices_, in embedding_bag_backward_cuda_sum_avg()
333 _embedding_bag_cuda(const Tensor &weight, const Tensor &indices_, in _embedding_bag_cuda()
510 const Tensor& indices_, in _embedding_bag_per_sample_weights_backward_cuda()
|
| H A D | FractionalMaxPool2d.cu | 162 auto indices_ = indices; in TORCH_IMPL_FUNC() local
239 auto indices_ = indices; in TORCH_IMPL_FUNC() local
|
| H A D | FractionalMaxPool3d.cu | 203 auto indices_ = indices; in fractional_max_pool3d_backward_out_cuda_template() local
267 auto indices_ = indices; in TORCH_IMPL_FUNC() local
|
| H A D | ScanUtils.cuh | 61 …kernel_scan_innermost_dim_with_indices(const scalar_t *self_, scalar_t *values_, int64_t *indices_, in tensor_kernel_scan_innermost_dim_with_indices()
155 …sor_kernel_scan_outer_dim_with_indices(const scalar_t *self_, scalar_t *values_, int64_t *indices_, in tensor_kernel_scan_outer_dim_with_indices()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | EmbeddingBag.cpp | 1180 const Tensor& indices_, in _embedding_bag_cpu_impl()
1320 const at::Tensor& indices_, in _embedding_bag_cpu_out()
1374 Tensor _embedding_bag_backward(const Tensor &grad, const Tensor &indices_, in _embedding_bag_backward()
1389 Tensor _embedding_bag_backward_symint(const Tensor &grad, const Tensor &indices_, in _embedding_bag_backward_symint()
1506 const Tensor& indices_, in _embedding_bag_dense_backward_cpu_sum_mean()
1594 Tensor _embedding_bag_dense_backward_cpu(const Tensor &grad_, const Tensor &indices_, in _embedding_bag_dense_backward_cpu()
1647 const Tensor& indices_, in _embedding_bag_per_sample_weights_backward_cpu_template()
|
| H A D | MaxUnpooling.cpp | 19 const Tensor& indices_, in max_unpooling2d_forward_out_cpu()
161 const Tensor& indices_, in max_unpooling3d_forward_out_cpu()
|
| H A D | Embedding.cpp | 76 const Tensor & grad_, const Tensor & indices_, int64_t num_weights, in embedding_sparse_backward()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_slice_grad_op_gpu.cu.cc | 70 const int64_t* __restrict__ indices_; member
85 const int64_t* indices_; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 45 auto indices_ = moveBatchDimToFront(indices, indices_bdim); in embedding_batch_rule() local
56 const Tensor& indices_, std::optional<int64_t> indices_bdim, in embedding_dense_backward_batch_rule()
|
| /aosp_15_r20/external/google-cloud-java/java-webrisk/proto-google-cloud-webrisk-v1/src/main/java/com/google/webrisk/v1/ |
| H A D | RawIndices.java | 72 private com.google.protobuf.Internal.IntList indices_; field in RawIndices
497 private com.google.protobuf.Internal.IntList indices_ = emptyIntList(); field in RawIndices.Builder
|
| /aosp_15_r20/external/google-cloud-java/java-webrisk/proto-google-cloud-webrisk-v1beta1/src/main/java/com/google/webrisk/v1beta1/ |
| H A D | RawIndices.java | 73 private com.google.protobuf.Internal.IntList indices_; field in RawIndices
499 private com.google.protobuf.Internal.IntList indices_ = emptyIntList(); field in RawIndices.Builder
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_reachability.h | 242 absl::flat_hash_map<uint64_t, int> indices_; variable
|
| H A D | indexed_array_analysis.h | 204 Array* indices_; variable
|
| /aosp_15_r20/cts/hostsidetests/adpf/app/hintsession/src/cpp/ |
| H A D | Model.h | 113 std::vector<Index> indices_; variable
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/ |
| H A D | stmt.h | 350 std::vector<ExprPtr> indices_; variable
881 std::vector<ExprPtr> indices_; variable
|
| H A D | registerizer.h | 178 std::vector<ExprPtr> indices_; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | sparse_to_dense_test.cc | 85 int indices_; member in tflite::__anon0fc1961c0111::SparseToDenseOpModel
|
| H A D | embedding_lookup_sparse_test.cc | 86 int indices_; member in tflite::__anon878db3e30111::EmbeddingLookupSparseOpModel
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/perception/ |
| H A D | max_pool_with_argmax_test.cc | 66 int indices_; member in tflite::ops::custom::__anon9efb64df0111::MaxpoolingWithArgMaxOpModel
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseTensorImpl.h | 30 Tensor indices_; // always a LongTensor member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | sharded_device_array.h | 324 pybind11::object indices_; variable
|