Searched refs:device_tensors (Results 1 – 2 of 2) sorted by relevance
124 std::deque<GraphMemory::LiveTensor>* device_tensors) { in FindOrCreateLiveTensor() argument134 device_tensors->push_front(temp); in FindOrCreateLiveTensor()135 live = &device_tensors->front(); in FindOrCreateLiveTensor()176 std::deque<LiveTensor>& device_tensors = in InferFromTrace() local183 node_stats.node_name(), i, &live_tensors, &device_tensors); in InferFromTrace()
207 device_tensors = dev0_tensors if i == 0 else dev1_tensors215 input_tensor = math_ops.cast(device_tensors[j], dtypes.float16)