Home
last modified time | relevance | path

Searched full:full_like (Results 1 – 25 of 128) sorted by relevance

123456

/external/pytorch/test/
Dtest_complex.py93 actual = torch.full_like(b, complex(2, 2))
100 actual = torch.full_like(b, complex(2, 2))
194 actual = torch.full_like(b, complex(2, 2))
218 actual = torch.full_like(b, complex(2, 2))
274 actual = torch.full_like(b, complex(2, 2))
281 actual = torch.full_like(b, complex(2, 2))
375 actual = torch.full_like(b, complex(2, 2))
399 actual = torch.full_like(b, complex(2, 2))
/external/pytorch/test/export/
Dtest_experimental.py234 …full_like = torch.ops.aten.full_like.default(div, 1, pin_memory = False, memory_format = torch.pre…
235 div_1 = torch.ops.aten.div.Scalar(full_like, 1); full_like = None
283 …full_like = torch.ops.aten.full_like.default(div, 1, pin_memory = False, memory_format = torch.pre…
284 div_1 = torch.ops.aten.div.Scalar(full_like, 1); full_like = None
/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/
Dinit.py116 torch.full_like: sharded_tensor.full,
132 ``torch.full_like``.
146 register_tensor_creation_op(torch.full_like)
/external/pytorch/torch/distributions/
Dgumbel.py41 torch.full_like(self.loc, finfo.tiny),
42 torch.full_like(self.loc, 1 - finfo.eps),
Dkumaraswamy.py53 torch.full_like(self.concentration0, 0),
54 torch.full_like(self.concentration0, 1),
Drelaxed_categorical.py90 log_scale = torch.full_like(
/external/executorch/backends/vulkan/runtime/graph/ops/impl/
DFull.cpp33 // size_or_in is IntListPtr when op is full and vTensorPtr if op is full_like
79 VK_REGISTER_OP(aten.full_like.default, full);
/external/pytorch/test/typing/pass/
Dcreation_ops.py93 # torch.full/full_like
95 torch.full_like(torch.full((2, 3), 3.141592), 2.71828)
/external/pytorch/test/distributed/
Dtest_functional_api.py529 expected.append(torch.full_like(tensor, (idx + 1)))
547 expected.append(torch.full_like(tensor, (idx + 1)))
563 expected.append(torch.full_like(tensor, (idx + 1)))
684 self.assertEqual(out, torch.full_like(t, 2.0))
689 self.assertEqual(t.grad, torch.full_like(t, 2.0))
835 self.assertEqual(out, torch.full_like(t, 2.0))
840 self.assertEqual(t.grad, torch.full_like(t, 2.0))
/external/executorch/backends/apple/mps/test/
Dtest_mps_linear.py334 scales, torch.full_like(scales, torch.finfo(torch.float32).eps)
336 zeros = torch.full_like(scales, 0)
366 scales, torch.full_like(scales, torch.finfo(torch.float32).eps)
369 zeros = torch.full_like(scales, 0)
/external/executorch/backends/qualcomm/partition/
Dcommon_defs.py22 exir_ops.edge.aten.full_like.default,
/external/pytorch/test/typing/reveal/
Dtensor_constructors.py95 # torch.full/full_like
97 reveal_type(torch.full_like(torch.full((2, 3), 3.141592), 2.71828)) # E: {Tensor}
/external/pytorch/torch/csrc/jit/passes/
Dremove_inplace_ops.cpp13 {aten::fill_, aten::full_like}};
16 // arguments of zeros_like and full_like so that the defaults are filled
/external/pytorch/aten/src/ATen/test/
Drng_test.h150 const auto expected = torch::full_like(actual, exp); in test_random_from_to()
206 const auto expected = torch::full_like(actual, exp); in test_random()
/external/executorch/extension/training/
DREADME.md167 …%full_like : [num_users=1] = call_function[target=torch.ops.aten.full_like.default](args = (%div, …
168 …%div_1 : [num_users=1] = call_function[target=torch.ops.aten.div.Tensor](args = (%full_like, %_to_…
173 …%full_like_1 : [num_users=1] = call_function[target=torch.ops.aten.full_like.default](args = (%_lo…
/external/executorch/exir/verification/
Dverifier.py107 torch.ops.aten.full_like.default, # TODO(T183507359)
279 torch.ops.aten.full_like.default, # TODO(T183507359)
/external/pytorch/tools/autograd/
Dgen_trace_type.py108 "fill": "full_like", # replacing aten::fill_ with aten::full_like
244 # `aten::zeros_like` and replacing `aten::fill_` with `aten::full_like`.
/external/executorch/backends/apple/mps/operators/
Dconstant_ops.py84 target = "aten.full_like.default"
/external/pytorch/torch/distributed/optim/
Dfunctional_rprop.py82 state["step_size"] = torch.full_like(gradient, lr)
Dfunctional_adagrad.py65 "sum": torch.full_like(p.data, initial_accumulator_value),
/external/executorch/docs/source/
Dextension-tensor.md310 Similarly to `empty()`, there are extra helper functions `full_like()`, `full_strided()`, `zeros_li…
385 | `at::full_like(tensor, value)` | `full_like(tensor, value)` |
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesFactory.cpp232 VMAP_SUPPORT(full_like, BASIC_UNARY_BATCH_RULE(ATEN_FN(full_like))); in TORCH_LIBRARY_IMPL()
/external/pytorch/torch/csrc/api/include/torch/optim/
Dadagrad.h86 state->sum(torch::full_like( in Optimizer()
/external/pytorch/docs/source/
Djit_unsupported.rst52 * :func:`torch.full_like`
/external/executorch/backends/vulkan/partitioner/
Dsupported_ops.py132 exir_ops.edge.aten.full_like.default,

123456