HomeSort by relevance Sort by last modified time
    Searched refs:computation_shape_array (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_config.py 112 computation_shape_array = np.asarray(computation_shape, dtype=np.int32)
115 if any(computation_shape_array < 1) or any(computation_shape_array > 2):
tpu_context.py 172 computation_shape_array = np.asarray(
174 num_cores_per_replica = np.prod(computation_shape_array)

Completed in 139 milliseconds