Searched refs:TensorReference (Results 1 – 13 of 13) 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()
609 friend class TensorReference; // For access to buf_ variable
165 TensorReference tensor_ref(tensor); in SetProtoFromGPU()237 TensorReference input_ref(*input); in DeviceToDeviceCopy()287 TensorReference input_ref(*gpu_tensor); in CopyGPUTensorToCPU()333 TensorReference input_ref(*cpu_tensor); in CopyCPUTensorToGPU()
109 v->push_back(TensorReference(buf)); in AddTensorReference()
181 TensorReference ref(*cpu_tensor); in CopyCPUTensorToDevice()244 TensorReference ref(*device_tensor); in CopyDeviceTensorToCPU()
80 TensorReference tensor_ref(out_of_line_values_on_host_); in Finalize()
178 TensorReference abnormal_detected_ref(abnormal_detected); in ComputeAsync()
326 std::vector<TensorReference> scratch_tensor_refs_;
303 TensorReference partition_ref(partition_count); in ComputeAsync()
808 TensorReference isvalid_dev_ref(isvalid_dev_tensor); in RunIfBoxIndexIsValid()