Home
last modified time | relevance | path

Searched full:source_tensor (Results 1 – 23 of 23) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_rma_local_test.cc85 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local
87 source_tensor.flat<float>()(i) = i / 2; in TEST_F()
90 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
95 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
109 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
128 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local
130 source_tensor.flat<float>()(i) = i / 2; in TEST_F()
133 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
140 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
154 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsubscribe.py164 source_tensor = tensor.op.inputs[0]
169 name_scope = source_tensor.op.name + '/subscription/'
172 outs += s(source_tensor)
207 source_tensor = tensor.op.inputs[0]
208 if prefix_name != source_tensor.op.name:
/aosp_15_r20/external/pytorch/docs/cpp/source/notes/
H A Dtensor_creation.rst304 tensor and does not occur in-place. For example, if we have a ``source_tensor``
309 torch::Tensor source_tensor = torch::randn({2, 3}, torch::kInt64);
315 torch::Tensor float_tensor = source_tensor.to(torch::kFloat32);
320 new memory, unrelated to the source ``source_tensor``.
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dtape.h635 auto source_tensor = sources_that_are_targets.find(id); in InitialGradients() local
636 if (source_tensor != sources_that_are_targets.end()) { in InitialGradients()
639 vspace.BuildOnesLike(source_tensor->second, &ones_like)); in InitialGradients()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNegateFallback.cpp24 m.impl("set_.source_Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
H A Dnative_functions.yaml7879 - func: set_.source_Tensor(Tensor(a!) self, Tensor source) -> Tensor(a!)
7885 autogen: set.source_Tensor, set.source_Tensor_out
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DConjugateFallback.cpp23 m.impl("set_.source_Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
H A DFunctionalizeFallbackKernel.cpp362 m.impl("set_.source_Tensor", TORCH_FN(set__functionalize)); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/
H A D_aten_impl.py56 elif op is torch.ops.aten.set_.source_Tensor:
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_has_decomposition.expect1153 aten::set.source_Tensor
1158 aten::set_.source_Tensor
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py6125 @register_lowering(torch.ops.aten.set_.source_Tensor)
6126 def set__source_tensor(self, source_tensor): argument
6128 source_tensor.realize()
6129 return TensorBox.create(ir.SetSourceTensorKernel(self, source_tensor))
6135 def fsdp_set_(self, source_tensor): argument
6137 source_tensor.realize()
6138 ir.SetSourceTensorKernel(self, source_tensor)
H A Dcomms.py517 torch.ops.aten.set_.source_Tensor,
H A Dir.py4945 python_kernel_name="torch.ops.aten.set_.source_Tensor",
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dmodel_ops.yaml321 aten::set_.source_Tensor: 1
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Dfunctional_utils.py406 torch.ops.aten.set_.source_Tensor,
/aosp_15_r20/external/tensorflow/tensorflow/python/feature_column/
H A Dfeature_column_v2.py2763 source_tensor = inputs.get(self.source_column)
2765 source_tensor,
2770 source_tensor = transformation_cache.get(self.source_column, state_manager)
2772 source_tensor,
H A Dfeature_column.py2416 source_tensor = inputs.get(self.source_column)
2418 source_tensor,
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfunctional_tensor.py577 and func is not torch.ops.aten.set_.source_Tensor
/aosp_15_r20/external/pytorch/torchgen/
H A Dgen_functionalization_type.py807 if g.inplace is not None and str(g.inplace.func.name) == "set_.source_Tensor":
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dtensor.py855 # aten::set_.source_Tensor(Tensor) gets special handling
/aosp_15_r20/external/pytorch/tools/autograd/templates/
H A Dpython_variable_methods.cpp1217 // aten::set_.source_Tensor(Tensor(a!) self, Tensor source) -> Tensor(a!) in THPVariable_set_()
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_pattern_matcher.py1422 check("call_function", torch.ops.aten.set_.source_Tensor, (t, t), {})
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_aotdispatch.py644 set_ = torch.ops.aten.set_.source_Tensor(primals_1, mul); primals_1 = set_ = None
1309 set_ = torch.ops.aten.set_.source_Tensor(primals_1, cat); primals_1 = set_ = None