Lines Matching refs:output_slot
91 // Records that "output_slot" of "node" has produced tensors of
93 void RecordSize(const Node* node, int output_slot, Bytes bytes);
96 Bytes TotalBytes(const Node* node, int output_slot) const;
99 // output_slot produced by one execution of "node".
100 Bytes SizeEstimate(const Node* node, int output_slot) const;
115 // the tensor generated by "output_slot" of "node". If
116 void RecordMaxMemorySize(const Node* node, int output_slot, Bytes bytes,
120 // Returns the maximum size in bytes of the tensor generated by "output_slot"
122 Bytes MaxMemorySize(const Node* node, int output_slot) const;
125 // generated by "output_slot" of "node".
127 int output_slot) const;
130 // generated by "output_slot" of "node".
131 DataType MaxMemoryType(const Node* node, int output_slot) const;
148 // Record the unique id of the tensor generated by "output_slot" of "node".
151 void RecordAllocationId(const Node* node, int output_slot, int64 alloc_id);
153 // Return the unique id of the tensor generated by "output_slot" of "node".
154 int64 AllocationId(const Node* node, int output_slot) const;
