Searched refs:tensor_alignment_ (Results 1 – 2 of 2) sorted by relevance
47 tensor_alignment_(tensor_alignment) {} in ArenaPlanner()322 arena_.Allocate(context_, tensor_alignment_, tensor.bytes, in CalculateAllocations()330 context_, tensor_alignment_, tensor.bytes, tensor_index, in CalculateAllocations()
133 int tensor_alignment_; variable