Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_rma_local_test.cc77 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local
79 source_tensor.flat<float>()(i) = i / 2; in TEST_F()
82 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
87 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
101 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
120 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local
122 source_tensor.flat<float>()(i) = i / 2; in TEST_F()
125 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
132 attr /*to_alloc_attr*/, &source_tensor, dev_locality, in TEST_F()
146 EXPECT_NE(DMAHelper::base(&source_tensor), DMAHelper::base(&sink_tensor)); in TEST_F()
/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py162 source_tensor = tensor.op.inputs[0]
167 name_scope = source_tensor.op.name + '/subscription/'
170 outs += s(source_tensor)
204 source_tensor = tensor.op.inputs[0]
205 if prefix_name != source_tensor.op.name:
/external/tensorflow/tensorflow/contrib/periodic_resample/kernels/
Dperiodic_resample_op.h271 const tensorflow::Tensor& source_tensor) { in do_periodic_resample_op() argument
272 const int rank = source_tensor.dims(); in do_periodic_resample_op()
312 auto input = source_tensor.flat<InputDataT>(); in do_periodic_resample_op()
/external/tensorflow/tensorflow/c/eager/
Dtape.h433 auto source_tensor = sources_that_are_targets.find(id); in InitialGradients() local
434 if (source_tensor != sources_that_are_targets.end()) { in InitialGradients()
435 (*result)[id].push_back(vspace.Ones(source_tensor->second)); in InitialGradients()
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py2865 source_tensor = inputs.get(self.source_column)
2867 source_tensor,
2872 source_tensor = transformation_cache.get(self.source_column, state_manager)
2874 source_tensor,
Dfeature_column.py2337 source_tensor = inputs.get(self.source_column)
2339 source_tensor,