Searched refs:TensorReferenceVector (Results 1 – 17 of 17) sorted by relevance
49 void QueueTensors(se::Stream* stream, TensorReferenceVector* tensors) { in QueueTensors()107 static void AddTensorReference(TensorReferenceVector* v, int64 size) { in AddTensorReference()121 TensorReferenceVector* v = nullptr; in TEST()126 v = new TensorReferenceVector; in TEST()137 v = new TensorReferenceVector; in TEST()158 TensorReferenceVector v; in TEST()175 TensorReferenceVector v; in TEST()194 TensorReferenceVector v1; in TEST()198 TensorReferenceVector v2; in TEST()217 TensorReferenceVector v; in TEST()[all …]
73 const TensorReferenceVector& tensors);121 TensorReferenceVector* mem;155 void QueueTensors(se::Stream* stream, TensorReferenceVector* tensors) in QueueTensors()191 TensorReferenceVector* accumulated_tensors_ GUARDED_BY(mu_);
101 accumulated_tensors_(new TensorReferenceVector), in EventMgr()164 const TensorReferenceVector& tensors) { in ThenDeleteTensors()187 accumulated_tensors_ = new TensorReferenceVector; in FlushAccumulatedTensors()
77 const TensorReferenceVector& tensor_refs) override;
602 DeviceContext* device_context, const TensorReferenceVector& tensor_refs) { in ConsumeListOfAccessedTensors()
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()
73 typedef gtl::InlinedVector<TensorReference, 4> TensorReferenceVector; typedef
338 TensorReferenceVector referenced_tensors; in TEST_F()363 TensorReferenceVector referenced_tensors; in TEST_F()
1160 void retrieve_accessed_tensors(TensorReferenceVector* out_vector);1542 TensorReferenceVector* out_vector) { in retrieve_accessed_tensors()
86 virtual void SetReferencedTensors(const TensorReferenceVector& tensors) = 0;115 void SetReferencedTensors(const TensorReferenceVector& tensors) override;
102 DeviceContext* context, const TensorReferenceVector& tensors) { in ConsumeListOfAccessedTensors()
109 DeviceContext* context, const TensorReferenceVector& tensors) override { in ConsumeListOfAccessedTensors()
161 const TensorReferenceVector& tensors) { in SetReferencedTensors()
119 const TensorReferenceVector& tensors) { in SetReferencedTensors()1704 TensorReferenceVector accessed_tensors; in Process()1778 TensorReferenceVector accessed; in Process()
87 void SetReferencedTensors(const TensorReferenceVector& tensors) override {} in SetReferencedTensors()