Home
last modified time | relevance | path

Searched refs:source_tensor (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local_test.cc84 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local
86 source_tensor.flat<float>()(i) = i / 2; in TEST_F()
89 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
94 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
108 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
127 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local
129 source_tensor.flat<float>()(i) = i / 2; in TEST_F()
132 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
139 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
153 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py168 source_tensor = tensor.op.inputs[0]
173 name_scope = source_tensor.op.name + '/subscription/'
176 outs += s(source_tensor)
211 source_tensor = tensor.op.inputs[0]
212 if prefix_name != source_tensor.op.name:
/external/tensorflow/tensorflow/c/eager/
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()
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py2696 source_tensor = inputs.get(self.source_column)
2698 source_tensor,
2703 source_tensor = transformation_cache.get(self.source_column, state_manager)
2705 source_tensor,
Dfeature_column.py2420 source_tensor = inputs.get(self.source_column)
2422 source_tensor,