Searched defs:edge_name (Results 1 – 4 of 4) sorted by relevance
52 Allocator* out_allocator, StringPiece edge_name, in CopyHostToDevice()200 void CopyTensor::ViaDMA(StringPiece edge_name, DeviceContext* send_dev_context, in ViaDMA()333 Allocator* out_allocator, StringPiece edge_name, in CopyDeviceToHost()
61 string edge_name(parsed.edge_name); in Send() local
65 StringPiece edge_name; member
218 NodeDef* input, const string& edge_name, in MaybeRewriteInput()