Searched refs:TensorReference (Results 1 – 12 of 12) sorted by relevance
30 class TensorReference {33 explicit TensorReference(const Tensor& tensor);35 ~TensorReference() {} in ~TensorReference()52 bool SharesBufferWith(const TensorReference& t) const { in SharesBufferWith()65 explicit TensorReference(TensorBuffer* test_buffer) : buf_(test_buffer) { in TensorReference() function73 typedef gtl::InlinedVector<TensorReference, 4> TensorReferenceVector;
55 bool operator()(const TensorReference& t1, in operator()56 const TensorReference& t2) const { in operator()62 size_t operator()(const TensorReference& t) const { return t.BufferHash(); } in operator()68 typedef std::unordered_set<TensorReference, TensorReferenceHashFn,
20 TensorReference::TensorReference(const Tensor& tensor) in TensorReference() function in tensorflow::TensorReference
40 const TensorReference tensor_ref(tensor); in Add()52 referenced_tensors_vector_.push_back(TensorReference(tensor)); in Add()
480 friend class TensorReference; // For access to buf_ variable
167 TensorReference tensor_ref(tensor); in SetProtoFromGPU()241 TensorReference input_ref(*input); in DeviceToDeviceCopy()291 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU()337 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU()
111 v->push_back(TensorReference(buf)); in AddTensorReference()
80 TensorReference tensor_ref(out_of_line_values_on_host_); in Finalize()
166 TensorReference abnormal_detected_ref(abnormal_detected); in ComputeAsync()
317 std::vector<TensorReference> scratch_tensor_refs_;
302 TensorReference partition_ref(partition_count); in ComputeAsync()
771 TensorReference isvalid_dev_ref(isvalid_dev_tensor); in RunIfBoxIndexIsValid()