| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | coordinate_descent_tuner.py | 121 def get_neighbour_values(self, name, orig_val, radius=1, include_self=False): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | ReduceUtils.h | 100 …_init(scalar_t* self_ptr, at::opmath_type<scalar_t>* buffer_ptr, int64_t size, bool include_self) { in _init()
|
| H A D | ScatterGatherKernel.cpp | 689 …uce_expanded_index(const Tensor& self, const Tensor& index, const Tensor& src, bool include_self) { in cpu_scatter_reduce_expanded_index()
860 const ReductionType& reduction, bool include_self) { in scatter_reduce_expanded_index_kernel()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_scatter_gather_ops.py | 87 unique_indices=True, include_self=True): argument
|
| /aosp_15_r20/external/bcc/tools/ |
| H A D | trace.py | 440 def generate_program(self, include_self): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | base_layer.py | 2850 def _flatten_layers(self, recursive=True, include_self=True): argument
2856 def _flatten_modules(self, recursive=True, include_self=True): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/utils/ |
| H A D | group_events.cc | 222 bool include_self) { in FindParentWithComparator()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | shim_common.cpp | 982 int32_t include_self) { in aoti_torch_scatter_reduce_out()
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | FunctionsManual.cpp | 6786 bool include_self, in scatter_reduce_jvp()
6819 bool include_self, in scatter_reduce_backward()
6918 bool include_self, in index_reduce_backward()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Indexing.cu | 1005 bool include_self, in index_reduce_func_cuda_impl()
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 5689 def meta_scatter_reduce_two(self, dim, index, src, reduce, include_self=True): argument
5695 def meta_scatter_reduce__two(self, dim, index, src, reduce, include_self=True): argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorAdvancedIndexing.cpp | 1066 bool include_self, in index_reduce_func_impl()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/ |
| H A D | generated_ops.cpp | 2179 const auto include_self = p_node->Input(5).toBool(); in __anon5d9c3eb9d302() local
|