Searched refs:allocation_info (Results 1 – 3 of 3) sorted by relevance
173 const Tensor& allocation_info = ctx->input(0); in Compute() local174 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(allocation_info.shape()), in Compute()179 allocation_proto.ParseFromString(allocation_info.scalar<string>()()), in Compute()
2261 Heap* heap, LinearAllocationArea allocation_info) in AllocateChunk() argument2262 : heap_(heap), allocation_info_(allocation_info) { in AllocateChunk()
2040 LocalAllocationBuffer(Heap* heap, LinearAllocationArea allocation_info);