Searched refs:TensorAllocatorFunc (Results 1 – 6 of 6) sorted by relevance
30 using TensorAllocatorFunc = std::function<Tensor*(const TensorShape& shape)>;62 TensorAllocatorFunc tensor_allocator) = 0;
75 bool ReadOutputNode(const string&, TensorAllocatorFunc) final;
113 TensorAllocatorFunc) { in ReadOutputNode() argument
207 TensorAllocatorFunc tensor_allocator) final { in ReadOutputNode()
50 TensorAllocatorFunc tensor_allocator) final;
355 const string& node_name, TensorAllocatorFunc tensor_allocator) { in ReadOutputNode()