Home
last modified time | relevance | path

Searched refs:max_partitions (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/variables/
Dpartitioned_variables_test.py195 def _testMinMaxVariablePartitioner(self, max_partitions, axis, min_slice_size, argument
199 max_partitions=max_partitions, axis=axis, min_slice_size=min_slice_size)
212 max_partitions=100,
223 max_partitions=100,
233 max_partitions=16,
241 max_partitions=1,
252 max_partitions=100,
260 max_partitions=100,
270 max_partitions=100,
278 max_partitions=100,
[all …]
/external/mesa3d/src/imagination/vulkan/
Dpvr_job_common.c319 uint32_t max_partitions; in pvr_setup_tiles_in_flight() local
367 max_partitions = PVR_GET_FEATURE_VALUE(dev_info, max_partitions, 0); in pvr_setup_tiles_in_flight()
369 partition_size * max_partitions); in pvr_setup_tiles_in_flight()
388 MIN2(max_partitions, usable_partition_size / partition_size); in pvr_setup_tiles_in_flight()
487 reg.partition_mask = (1 << max_partitions) - 1; in pvr_setup_tiles_in_flight()
/external/tensorflow/tensorflow/python/ops/
Dpartitioned_variables.py156 def min_max_variable_partitioner(max_partitions=1, axis=0, argument
214 max_partitions,
/external/mesa3d/src/imagination/common/device_info/
Daxe-1-16m.h81 .max_partitions = 4U,
Dbxs-4-64.h93 .max_partitions = 16U,
Dgx6250.h84 .max_partitions = 8U,
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils_test.py69 max_partitions=5, axis=0, min_slice_size=8 << 10))
296 max_partitions=5, axis=0, min_slice_size=8 << 10))
306 max_partitions=5, axis=0, min_slice_size=16 << 10))
334 max_partitions=5, axis=0, min_slice_size=8 << 10))
/external/mesa3d/src/imagination/vulkan/winsys/pvrsrvkm/
Dpvr_srv.c478 uint32_t max_partitions = PVR_GET_FEATURE_VALUE(dev_info, max_partitions, 0); in pvr_srv_get_total_reserved_partition_size() local
481 return tile_size_x * tile_size_y * max_partitions * in pvr_srv_get_total_reserved_partition_size()
487 return (uint64_t)max_partitions * 1024U; in pvr_srv_get_total_reserved_partition_size()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.h426 int max_partitions,
Dnnapi_delegate.cc6611 int max_partitions, in LimitDelegatedPartitions() argument
6615 if (max_partitions <= 0 || num_partitions <= max_partitions) { in LimitDelegatedPartitions()
6627 if (number_delegated_partitions > max_partitions) { in LimitDelegatedPartitions()
6638 for (int i = 0; i < max_partitions; i++) { in LimitDelegatedPartitions()
/external/mesa3d/src/imagination/common/
Dpvr_device_info.h317 uint32_t max_partitions; member
/external/trusty/arm-trusted-firmware/services/std_svc/spm/el3_spmc/
Dspmc_main.c965 uint32_t max_partitions, in partition_info_get_handler_v1_1() argument
977 if (*partition_count >= max_partitions) { in partition_info_get_handler_v1_1()
1010 if (*partition_count >= max_partitions) { in partition_info_get_handler_v1_1()
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py163 max_partitions=self._max_shards,
/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/
Dembedding_ops_test.py178 max_partitions=num_shards, min_slice_size=1),
/external/googleapis/google/spanner/v1/
Dspanner.proto923 int64 max_partitions = 2; field
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1701 …argspec: "args=[\'max_partitions\', \'axis\', \'min_slice_size\', \'bytes_per_string_element\'], v…