| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local_test.cc | 85 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 …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe.py | 164 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:
|
| /external/pytorch/docs/cpp/source/notes/ |
| D | tensor_creation.rst | 304 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``.
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NegateFallback.cpp | 24 m.impl("set_.source_Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|
| D | native_functions.yaml | 7879 - func: set_.source_Tensor(Tensor(a!) self, Tensor source) -> Tensor(a!)
7885 autogen: set.source_Tensor, set.source_Tensor_out
|
| /external/pytorch/aten/src/ATen/ |
| D | ConjugateFallback.cpp | 23 m.impl("set_.source_Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|
| D | FunctionalizeFallbackKernel.cpp | 362 m.impl("set_.source_Tensor", TORCH_FN(set__functionalize)); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/test/cpp_extensions/open_registration_extension/pytorch_openreg/ |
| D | _aten_impl.py | 56 elif op is torch.ops.aten.set_.source_Tensor:
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 635 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()
|
| /external/pytorch/test/mobile/model_test/ |
| D | model_ops.yaml | 321 aten::set_.source_Tensor: 1
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_has_decomposition.expect | 1153 aten::set.source_Tensor
1158 aten::set_.source_Tensor
|
| /external/pytorch/torch/_inductor/ |
| D | comms.py | 517 torch.ops.aten.set_.source_Tensor,
|
| D | lowering.py | 6125 @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)
|
| D | ir.py | 4945 python_kernel_name="torch.ops.aten.set_.source_Tensor",
|
| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | functional_utils.py | 406 torch.ops.aten.set_.source_Tensor,
|
| /external/pytorch/torch/_subclasses/ |
| D | functional_tensor.py | 577 and func is not torch.ops.aten.set_.source_Tensor
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 2763 source_tensor = inputs.get(self.source_column)
2765 source_tensor,
2770 source_tensor = transformation_cache.get(self.source_column, state_manager)
2772 source_tensor,
|
| D | feature_column.py | 2416 source_tensor = inputs.get(self.source_column)
2418 source_tensor,
|
| /external/pytorch/torchgen/ |
| D | gen_functionalization_type.py | 807 if g.inplace is not None and str(g.inplace.func.name) == "set_.source_Tensor":
|
| /external/pytorch/torch/_dynamo/variables/ |
| D | tensor.py | 855 # aten::set_.source_Tensor(Tensor) gets special handling
|
| /external/pytorch/tools/autograd/templates/ |
| D | python_variable_methods.cpp | 1217 // aten::set_.source_Tensor(Tensor(a!) self, Tensor source) -> Tensor(a!) in THPVariable_set_()
|
| /external/pytorch/test/inductor/ |
| D | test_pattern_matcher.py | 1422 check("call_function", torch.ops.aten.set_.source_Tensor, (t, t), {})
|
| /external/pytorch/test/functorch/ |
| D | test_aotdispatch.py | 644 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
|