Searched refs:allocate_count (Results 1 – 2 of 2) sorted by relevance
678 int64 allocate_count = in AllocateBytes() local681 tf_data_type, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()732 int64 allocate_count = in AllocateBytes() local737 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()
373 int64 allocate_count = in AllocateBytes() local376 tf_data_type, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()417 int64 allocate_count = in AllocateBytes() local422 output_index_, TensorShape({allocate_count}), &temporary_memory)); in AllocateBytes()