| /external/pytorch/test/ |
| D | test_scatter_gather_ops.py | 87 unique_indices=True, include_self=True): argument
106 … actual = fn(base.clone(), dim, idx, src, reduce=reduction, include_self=include_self)
113 counts = torch.zeros(base.shape, dtype=torch.long, device=device) + include_self
127 if ((not include_self) and counts[tuple(ii)] == 0):
216 for include_self in (True, False):
221 include_self=include_self)
226 for include_self in (True, False):
229 include_self=include_self)
234 for include_self in (True, False):
239 include_self=include_self)
[all …]
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | ReduceUtils.h | 91 inline void init(scalar_t* out, int64_t size, bool include_self = false) {
93 if (!include_self) {
100 …_init(scalar_t* self_ptr, at::opmath_type<scalar_t>* buffer_ptr, int64_t size, bool include_self) { in _init() argument
101 if (!include_self) { in _init()
102 init<at::opmath_type<scalar_t>, reduce>(buffer_ptr, size, include_self); in _init()
|
| 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() argument
794 _init<scalar_t, reduce>(self_ptr, buffer_ptr, K, include_self); in cpu_scatter_reduce_expanded_index()
806 int64_t count = include_self ? 1 : 0; in cpu_scatter_reduce_expanded_index()
860 const ReductionType& reduction, bool include_self) { in scatter_reduce_expanded_index_kernel() argument
864 cpu_scatter_reduce_expanded_index<scalar_t, reduce>(self, index, src, include_self); in scatter_reduce_expanded_index_kernel()
|
| /external/pytorch/torch/_inductor/runtime/ |
| D | coordinate_descent_tuner.py | 121 def get_neighbour_values(self, name, orig_val, radius=1, include_self=False): argument
157 if include_self:
187 include_self=True,
|
| /external/pytorch/torch/csrc/api/src/nn/ |
| D | module.cpp | 181 std::vector<std::shared_ptr<Module>> Module::modules(bool include_self) const { in modules()
183 if (include_self) { in modules()
198 bool include_self) const { in named_modules()
200 if (include_self) { in named_modules()
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset16.py | 128 include_self: bool,
134 if not include_self:
|
| /external/pytorch/torch/csrc/api/include/torch/nn/ |
| D | module.h | 237 std::vector<std::shared_ptr<Module>> modules(bool include_self = true) const;
254 bool include_self = true) const;
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorAdvancedIndexing.cpp | 258 bool include_self) { in TORCH_META_FUNC2()
259 (void) include_self; in TORCH_META_FUNC2()
391 bool include_self) { in TORCH_PRECOMPUTE_META_FUNC()
392 (void)include_self; in TORCH_PRECOMPUTE_META_FUNC()
1066 bool include_self, in index_reduce_func_impl() argument
1070 if (!include_self) { in index_reduce_func_impl()
1147 auto counts = include_self ? at::ones_like(result) : at::zeros_like(result); in index_reduce_func_impl()
1159 auto counts = include_self ? at::ones_like(result) : at::zeros_like(result); in index_reduce_func_impl()
1922 bool include_self, in TORCH_IMPL_FUNC()
1934 scatter_reduce_expanded_index_stub(self.device().type(), out, index, src, op, include_self); in TORCH_IMPL_FUNC()
[all …]
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Indexing.cu | 1005 bool include_self, in index_reduce_func_cuda_impl() argument
1021 if (!include_self) { in index_reduce_func_cuda_impl()
1183 bool include_self, in TORCH_IMPL_FUNC()
1188 …index_reduce_func_cuda_impl(self, dim, index, source, include_self, ReductionType::PROD, reduce_mu… in TORCH_IMPL_FUNC()
1190 …index_reduce_func_cuda_impl(self, dim, index, source, include_self, ReductionType::MEAN, reduce_ad… in TORCH_IMPL_FUNC()
1191 auto counts = include_self ? at::ones_like(result) : at::zeros_like(result); in TORCH_IMPL_FUNC()
1200 …index_reduce_func_cuda_impl(self, dim, index, source, include_self, ReductionType::MAX, reduce_max… in TORCH_IMPL_FUNC()
1202 …index_reduce_func_cuda_impl(self, dim, index, source, include_self, ReductionType::MIN, reduce_min… in TORCH_IMPL_FUNC()
|
| /external/pytorch/torch/_inductor/ |
| D | decomposition.py | 887 include_self: bool = True,
894 if include_self:
928 include_self=include_self,
|
| /external/bcc/tools/ |
| D | trace.py | 440 def generate_program(self, include_self): argument
453 elif not include_self:
849 self.args.include_self)
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | utils.py | 120 …return list(obj._flatten_layers(include_self=False, recursive=False)) # pylint: disable=protected…
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | models.py | 320 for layer in model._flatten_layers(include_self=False, recursive=False):
514 model._flatten_layers(include_self=False, recursive=False))
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer.py | 2844 nested_layers = self._flatten_modules(include_self=False, recursive=False)
2850 def _flatten_layers(self, recursive=True, include_self=True): argument
2852 recursive=recursive, include_self=include_self):
2856 def _flatten_modules(self, recursive=True, include_self=True): argument
2866 if include_self:
|
| D | base_layer_v1.py | 2114 layers = self._flatten_layers(include_self=False, recursive=False)
2124 layers = self._flatten_layers(include_self=False, recursive=False)
|
| D | training_v1.py | 506 list(self._flatten_layers(include_self=False, recursive=False))))
1716 layers = self._flatten_layers(include_self=False, recursive=False)
|
| /external/tensorflow/tensorflow/core/profiler/utils/ |
| D | group_events.cc | 222 bool include_self) { in FindParentWithComparator() argument
225 if (include_self) { in FindParentWithComparator()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 6786 bool include_self, in scatter_reduce_jvp() argument
6790 return at::scatter_reduce(self_t, dim, index, src_t, reduce, include_self); in scatter_reduce_jvp()
6802 dim, index, mask_src.to(self_t.dtype()), "sum", include_self); in scatter_reduce_jvp()
6804 .scatter_reduce(dim, index, masked_src_t, "sum", include_self) in scatter_reduce_jvp()
6819 bool include_self, in scatter_reduce_backward() argument
6838 masked_self.scatter_reduce(dim, index, src, reduce, include_self); in scatter_reduce_backward()
6851 self.scatter_reduce(dim, index, masked_src, reduce, include_self); in scatter_reduce_backward()
6868 Tensor N = include_self ? ones_like(grad) : zeros_like(grad); in scatter_reduce_backward()
6891 if (!include_self) { in scatter_reduce_backward()
6918 bool include_self, in index_reduce_backward() argument
[all …]
|
| D | FunctionsManual.h | 1050 bool include_self,
1060 bool include_self,
1074 bool include_self,
|
| /external/pytorch/torch/csrc/inductor/aoti_torch/generated/ |
| D | c_shim_cpu.h | 75 …sorHandle index, AtenTensorHandle source, const char* reduce, int32_t include_self, AtenTensorHand…
118 …t64_t dim, AtenTensorHandle index, AtenTensorHandle src, const char* reduce, int32_t include_self);
|
| D | c_shim_cuda.h | 84 …sorHandle index, AtenTensorHandle source, const char* reduce, int32_t include_self, AtenTensorHand…
127 …t64_t dim, AtenTensorHandle index, AtenTensorHandle src, const char* reduce, int32_t include_self);
|
| /external/pytorch/torch/masked/ |
| D | _ops.py | 717 0, scatter_indices, new_values, reduce=reduce, include_self=False
723 0, scatter_indices, new_values, reduce=reduce, include_self=False
776 0, scatter_indices, values, reduce, include_self=False
|
| /external/pytorch/torch/csrc/inductor/aoti_torch/ |
| D | shim_common.cpp | 982 int32_t include_self) { in aoti_torch_scatter_reduce_out() argument
995 (bool)include_self); in aoti_torch_scatter_reduce_out()
|
| /external/pytorch/torch/csrc/inductor/aoti_torch/c/ |
| D | shim.h | 551 int32_t include_self);
|
| /external/crosvm/acpi_tables/src/ |
| D | aml.rs | 326 fn insert_length(aml: &mut Vec<u8>, position: usize, len: usize, include_self: bool) { in insert_length()
338 let length = len + if include_self { length_length } else { 0 }; in insert_length()
|