Searched refs:tensor_ref (Results 1 – 5 of 5) sorted by relevance
40 const TensorReference tensor_ref(tensor); in Add() local41 if (!referenced_tensors_set_->insert(tensor_ref).second) { in Add()43 tensor_ref.Unref(); in Add()
80 TensorReference tensor_ref(out_of_line_values_on_host_); in Finalize()90 stream, [tensor_ref]() { tensor_ref.Unref(); }); in Finalize()
173 for (auto tensor_ref : scratch_tensor_refs_) { in ~CudaSolver() local174 tensor_ref.Unref(); in ~CudaSolver()
165 TensorReference tensor_ref(tensor); in SetProtoFromGPU() local168 total_bytes, alloc, tensor_ref]() { in SetProtoFromGPU()172 tensor_ref.Unref(); in SetProtoFromGPU()
259 tensor_ref = array_ops.zeros([5, 2, 3])261 vbn = virtual_batchnorm.VBN(tensor_ref, name='vbn7', batch_axis=1)