Searched refs:allocate_count (Results 1 – 1 of 1) sorted by relevance
374 int64 allocate_count = in AllocateBytes() local377 tf_data_type, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()419 int64 allocate_count = in AllocateBytes() local424 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()