Searched defs:source_tensor (Results 1 – 3 of 3) sorted by relevance
85 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local128 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local218 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local239 Tensor source_tensor(DT_FLOAT, TensorShape({8})); in TEST_F() local
635 auto source_tensor = sources_that_are_targets.find(id); in InitialGradients() local
6126 def set__source_tensor(self, source_tensor): argument6135 def fsdp_set_(self, source_tensor): argument