Searched refs:tensor_flat (Results 1 – 1 of 1) sorted by relevance
86 auto tensor_flat = tensor->flat<T>(); in SetRandomValuesForMaxPooling() local91 tensor_flat(i) = cur; in SetRandomValuesForMaxPooling()99 T tmp = tensor_flat(i); in SetRandomValuesForMaxPooling()100 tensor_flat(i) = tensor_flat(j); in SetRandomValuesForMaxPooling()101 tensor_flat(j) = tmp; in SetRandomValuesForMaxPooling()