Searched refs:TensorReferenceVector (Results 1 – 18 of 18) sorted by relevance
32 TensorReferenceVector tensors; in TEST()55 TensorReferenceVector tensors; in TEST()91 TensorReferenceVector tensors; in TEST()116 TensorReferenceVector tensors; in TEST()
45 void FreezeAndReturnReferences(TensorReferenceVector* out_vector);66 TensorReferenceVector referenced_tensors_vector_;
24 TensorReferenceVector refs; in ~UniqueTensorReferences()70 TensorReferenceVector* out_vector) { in FreezeAndReturnReferences()
76 typedef gtl::InlinedVector<TensorReference, 4> TensorReferenceVector; typedef
1212 void retrieve_accessed_tensors(TensorReferenceVector* out_vector);1641 TensorReferenceVector* out_vector) { in retrieve_accessed_tensors()
379 TensorReferenceVector referenced_tensors; in TEST_F()403 TensorReferenceVector referenced_tensors; in TEST_F()
71 const TensorReferenceVector& tensors);121 TensorReferenceVector* mem;157 void QueueTensors(se::Stream* stream, TensorReferenceVector* tensors) in QueueTensors()193 TensorReferenceVector* accumulated_tensors_ GUARDED_BY(mu_);
66 void QueueTensors(se::Stream* stream, TensorReferenceVector* tensors) { in QueueTensors()124 static void AddTensorReference(TensorReferenceVector* v, int64 size) { in AddTensorReference()138 TensorReferenceVector* v = nullptr; in TEST()143 v = new TensorReferenceVector; in TEST()154 v = new TensorReferenceVector; in TEST()175 TensorReferenceVector v; in TEST()192 TensorReferenceVector v; in TEST()211 TensorReferenceVector v1; in TEST()215 TensorReferenceVector v2; in TEST()234 TensorReferenceVector v; in TEST()[all …]
101 accumulated_tensors_(new TensorReferenceVector), in EventMgr()164 const TensorReferenceVector& tensors) { in ThenDeleteTensors()187 accumulated_tensors_ = new TensorReferenceVector; in FlushAccumulatedTensors()
72 const TensorReferenceVector& tensor_refs) override;
545 DeviceContext* device_context, const TensorReferenceVector& tensor_refs) { in ConsumeListOfAccessedTensors()
86 virtual void SetReferencedTensors(const TensorReferenceVector& tensors) = 0;116 void SetReferencedTensors(const TensorReferenceVector& tensors) override;
99 DeviceContext* context, const TensorReferenceVector& tensors) { in ConsumeListOfAccessedTensors()
142 DeviceContext* context, const TensorReferenceVector& tensors) override { in ConsumeListOfAccessedTensors()
169 const TensorReferenceVector& tensors) { in SetReferencedTensors()
125 const TensorReferenceVector& tensors) { in SetReferencedTensors()1795 TensorReferenceVector accessed_tensors; in Process()1868 TensorReferenceVector accessed; in Process()
65 TensorReferenceVector protected_tensors_;
93 void SetReferencedTensors(const TensorReferenceVector& tensors) override {} in SetReferencedTensors()