Searched defs:chunk_dim (Results 1 – 3 of 3) sorted by relevance
38 def __init__(self, chunk_dim=0, device_count=0): argument
408 int chunk_dim) { in CalculateCacheFriendlyTasksCount()
3304 def lstm_cell(inp, hx, cx, hh_weight, hh_bias, hr_weight, chunk_dim): argument