Searched refs:_dims (Results 1 – 4 of 4) sorted by relevance
762 self._dims = [Dimension(d) for d in dims]764 self._dims = None767 self._dims = None769 self._dims = [775 self._dims = dims.dims781 self._dims = [as_dimension(dims)]783 self._dims = []786 self._dims.append(as_dimension(d))804 if self._dims is not None:805 return "TensorShape(%r)" % [dim.value for dim in self._dims][all …]
251 : _device(device), _context(context), _queue(queue), _dims(dims) in TestNonUniformWorkGroup()314 for (i = _dims; i < MAX_DIMS; i++) in TestNonUniformWorkGroup()327 for (i = 0; i < _dims; i++) in TestNonUniformWorkGroup()341 for (i = 0; i < _dims; i++) in TestNonUniformWorkGroup()349 for (i = 0; i < _dims; i++) in TestNonUniformWorkGroup()353 for (i = _dims; i < MAX_DIMS; i++) in TestNonUniformWorkGroup()387 for (i = 0; i < _dims; i++) { in setLocalWorkgroupSize()400 for (i = 0; i < _dims; i++) { in setGlobalWorkgroupSize()483 _referenceRegionArray[i].get_work_dim = _dims; in calculateExpectedValues()535 for (i = 0; i < _dims; i++) { in prepareDevice()[all …]
99 const cl_uint _dims; variable
91 self._dims = dims92 self._output_size = tensor_shape.TensorShape(self._dims)93 self._state_size = (tensor_shape.TensorShape(self._dims),94 tensor_shape.TensorShape(self._dims))