| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | mirrored_strategy_test.py | 78 def testMinimizeLoss(self, distribution): argument
84 def testReplicaId(self, distribution): argument
87 def testNumReplicasInSync(self, distribution): argument
90 def testCallAndMergeExceptions(self, distribution): argument
93 def testRunRegroupError(self, distribution): argument
105 def testReduceToCpu(self, distribution): argument
115 def testReduceToCpuNested(self, distribution): argument
132 def reduce_axis_helper(self, distribution, replica_squared_fn): argument
146 def testReduceAxisToCpu(self, distribution): argument
164 def testReduceAxisToCpuUnknownShape(self, distribution): argument
[all …]
|
| H A D | vars_test.py | 85 def testAssign(self, distribution, experimental_run_tf_function): argument
124 def testAssignOnWriteVar(self, distribution, experimental_run_tf_function): argument
168 def testAssignPerReplicaVal(self, distribution, experimental_run_tf_function): argument
225 def testValueInReplicaContext(self, distribution): argument
242 def testValueInReplicaContextAssignDirectValue(self, distribution, argument
260 def testReadValueInReplicaContext(self, distribution, argument
284 def testReadValueInCrossReplicaContext(self, distribution, argument
310 def testAssignOutOfScope(self, distribution): argument
319 def testInitializedToSameValueInsideEagerRun(self, distribution): argument
341 def testAggregationOnlyFirstReplica(self, distribution): argument
[all …]
|
| H A D | custom_training_loop_input_test.py | 82 def testConstantNumpyInput(self, distribution): argument
103 def testStatefulExperimentalRunAlwaysExecute(self, distribution): argument
124 def testFullEager(self, distribution): argument
141 def testGetNextAsOptional(self, distribution): argument
161 def testGetNextAsOptionalExampleUsage(self, distribution): argument
186 def testFullEagerTPU(self, distribution): argument
203 def testStepInFunction(self, distribution): argument
223 def testRunInFunction(self, distribution): argument
249 def testNestedOutput(self, distribution): argument
278 def testRunInFunctionAutoGraphApplication(self, distribution): argument
[all …]
|
| H A D | values_test.py | 76 def testMakeDistributedValueFromTensor(self, distribution): argument
96 def testMakeDistributedValueSingleNumpyArrayConstant(self, distribution): argument
116 def testMakeDistributedValueTupleConstant(self, distribution): argument
139 def testMakeDistributedValueNestedStructurePerReplica(self, distribution): argument
165 def testMakeDistributedValueSpareTensor(self, distribution): argument
187 def testMakeDistributedValueExtractFromArray(self, distribution): argument
205 def testMakeDistributedValueAndRun(self, distribution): argument
241 def testMakeDistributedValueDefaultDevicePlacement(self, distribution): argument
265 def testMakeDistributedValueExplicitDevicePlacement(self, distribution, argument
297 def testUsePerReplicaInvalidContextGivesError(self, distribution): argument
[all …]
|
| H A D | one_device_strategy_test.py | 40 def testMinimizeLoss(self, distribution): argument
46 def testReplicaId(self, distribution): argument
49 def testCallAndMergeExceptions(self, distribution): argument
52 def testReplicateDataset(self, distribution): argument
64 def testMakeInputFnIteratorWithDataset(self, distribution): argument
76 def testMakeInputFnIteratorWithCallable(self, distribution): argument
92 def testNumpyDataset(self, distribution): argument
95 def testRun(self, distribution): argument
98 def testAllReduceSum(self, distribution): argument
101 def testAllReduceSumGradients(self, distribution): argument
[all …]
|
| H A D | mirrored_variable_test.py | 103 def testVariableInFuncGraph(self, distribution): argument
117 def testVariableWithTensorInitialValueInFunction(self, distribution): argument
137 def testSingleVariable(self, distribution): argument
151 def testUnnamedVariable(self, distribution): argument
162 def testMultipleVariables(self, distribution): argument
176 def testMultipleVariablesWithSameCanonicalName(self, distribution): argument
197 def testVariableWithSameCanonicalNameAcrossThreads(self, distribution): argument
211 def testWithVariableAndVariableScope(self, distribution): argument
250 def testWithGetVariableAndVariableScope(self, distribution): argument
289 def testOnlyFirstReplicaUpdatesVariables(self, distribution): argument
[all …]
|
| H A D | metrics_v1_test.py | 95 def _test_metric(self, distribution, dataset_fn, metric_fn, expected_fn): argument
137 def testMean(self, distribution): argument
149 def testAccuracy(self, distribution): argument
164 def testMeanPerClassAccuracy(self, distribution): argument
183 def testMeanIOU(self, distribution): argument
201 def testMeanTensor(self, distribution): argument
220 def testAUCROC(self, distribution): argument
234 def testAUCPR(self, distribution): argument
248 def testFalseNegatives(self, distribution): argument
261 def testFalseNegativesAtThresholds(self, distribution): argument
[all …]
|
| H A D | input_lib_test.py | 290 def testMultiDeviceIterInitialize(self, distribution): argument
321 def testOneDeviceCPU(self, input_type, api_type, iteration_type, distribution, argument
345 distribution, enable_get_next_as_optional): argument
371 distribution, enable_get_next_as_optional): argument
394 def testTPU(self, input_type, api_type, iteration_type, distribution, argument
425 def testTupleDataset(self, input_type, api_type, iteration_type, distribution, argument
461 distribution, enable_get_next_as_optional): argument
494 def testIterableIterator(self, distribution): argument
513 def testIterableIteratorError(self, distribution): argument
559 drop_remainder, distribution): argument
[all …]
|
| H A D | input_lib_type_spec_test.py | 56 def testTypeSpec(self, input_type, distribution, argument
89 distribution, enable_get_next_as_optional): argument
124 def testDoesNotTriggerFunctionTracing(self, input_type, distribution, argument
174 def testInputSignatureForPerReplicaValues(self, distribution, argument
208 def testInputSignatureForNestedPerReplicaValues(self, distribution): argument
231 def testMostSpecificCompatibleType(self, input_type, distribution, argument
270 self, distribution, enable_get_next_as_optional, argument
339 def testTypeSpecBase(self, distribution, enable_get_next_as_optional): argument
366 def testTypeSpecReturnedFromTFFunction(self, distribution, argument
414 def testTypeSpecRaggedTensor(self, distribution, enable_get_next_as_optional): argument
[all …]
|
| H A D | distributed_variable_test.py | 100 def testExtendsVariable(self, distribution, synchronization, aggregation): argument
106 def testCheckpointing(self, distribution, synchronization, aggregation, mode): argument
139 def testTraceback(self, distribution, synchronization, aggregation): argument
158 def testSelectReplica(self, distribution, synchronization, aggregation): argument
164 def testIsTensorLike(self, distribution, synchronization, aggregation): argument
177 def testAssignReturnValueIsTensorLike(self, distribution, synchronization, argument
207 def testDeepCopy(self, distribution, synchronization, aggregation): argument
242 def testAssignSignature(self, distribution, synchronization, aggregation): argument
271 def testStrategyExtendedUpdate(self, distribution, synchronization, argument
301 def testSaveNonDistributed(self, distribution, synchronization, aggregation): argument
[all …]
|
| H A D | moving_averages_test.py | 53 def testReplicaModeWithoutZeroDebias(self, distribution): argument
80 def testReplicaMode(self, distribution): argument
102 def testCrossDeviceWithoutZeroDebias(self, distribution): argument
131 def testCrossDevice(self, distribution): argument
146 def testUpdateContext(self, distribution, use_function): argument
164 def testAssignVariable(self, distribution): argument
188 def testReplicaContextEager(self, distribution, use_function): argument
215 def testCrossReplicaContextEager(self, distribution, use_function): argument
245 def testReplicaContextGraph(self, distribution): argument
262 def testCrossReplicaContextGraph(self, distribution): argument
|
| H A D | mirrored_values_test.py | 38 def _make_mirrored(distribution=None): argument
123 def _save_mirrored(self, distribution): argument
168 def _restore_mirrored(self, save_path, distribution): argument
183 def testSaveAndRestoreMirroredOneGraph(self, distribution): argument
202 def testSaveMirroredRestoreMirrored(self, distribution): argument
215 def testSaveMirroredRestoreNormal(self, distribution): argument
228 def testSaveNormalRestoreMirrored(self, distribution): argument
|
| H A D | remote_mirrored_strategy_eager_test.py | 51 def testNumReplicasInSync(self, distribution): argument
54 def testMinimizeLoss(self, distribution): argument
57 def testDeviceScope(self, distribution): argument
60 def testMakeInputFnIteratorWithDataset(self, distribution): argument
63 def testMakeInputFnIteratorWithCallable(self, distribution): argument
|
| H A D | tf_function_test.py | 53 self, distribution, run_functions_eagerly): argument
79 self, distribution, run_functions_eagerly): argument
109 def testReadVariableInsideFunction(self, distribution, run_functions_eagerly): argument
151 def testRetraceOnSavingFirstTraceInScope(self, distribution): argument
176 def testRetraceOnSavingFirstTraceOutsideScope(self, distribution): argument
|
| H A D | strategy_common_test.py | 99 def testLocalResultForDictionary(self, distribution): argument
110 def testLocalResultForList(self, distribution): argument
121 def testLocalResultForTuple(self, distribution): argument
133 def testLocalResultForNestedStruct(self, distribution): argument
162 def testLocalResultForNestedStructWithoutTensor(self, distribution): argument
174 def testLocalResultForScalarValue(self, distribution): argument
187 def testLocalResultForDictionaryDifferentReplicas(self, distribution): argument
203 def testLocalResultForTensor(self, distribution): argument
|
| H A D | combinations_test.py | 45 def testClusterParams(self, distribution, has_chief, num_workers): argument
54 def testClusterParamsHasDefault(self, distribution, has_chief, num_workers): argument
72 def testClusterParamsAreOptional(self, distribution): argument
148 def testMultiWorkerCanFail(self, distribution): argument
169 def test(self, distribution): argument
|
| H A D | step_fn.py | 24 def __init__(self, distribution): argument
28 def distribution(self): member in Step
49 def __init__(self, dataset_fn, distribution): argument
84 def __init__(self, dataset_fn, loss_fn, optimizer, distribution, argument
|
| H A D | zero_batch_test.py | 45 def testBNWithZeroBatchInputGraph(self, distribution, fused): argument
116 def testBNWithZeroBatchInput(self, distribution, fused): argument
166 def testBNWithDynamicBatchInputEager(self, distribution, fused): argument
|
| H A D | custom_training_loop_gradient_test.py | 69 def testStepInFunctionGradient(self, distribution): argument
95 def testRunInFunctionGradient(self, distribution): argument
123 def testNestedFunction(self, distribution, model_in_tf_function): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_loss_scaling_utilities_test.py | 73 def testComputeAverageLossDefaultGlobalBatchSize(self, distribution): argument
92 def testComputeAverageLossSampleWeights(self, distribution): argument
133 def testComputeAverageLossDtype(self, distribution): argument
171 def testComputeAverageLossInCrossReplicaContext(self, distribution): argument
184 def testScaleRegularizationLoss(self, distribution): argument
203 def testScaleRegularizationLossInCrossReplicaContext(self, distribution): argument
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/CPP/ |
| H A D | DFT.cpp | 81 std::uniform_real_distribution<float> distribution(-5.f, 5.f); in TEST_SUITE() local
99 std::uniform_real_distribution<float> distribution(-5.f, 5.f); in DATA_TEST_CASE() local
117 std::uniform_real_distribution<float> distribution(-5.f, 5.f); in TEST_SUITE() local
135 std::uniform_real_distribution<float> distribution(-5.f, 5.f); in DATA_TEST_CASE() local
152 std::uniform_real_distribution<float> distribution(-1.f, 1.f); in TEST_SUITE() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/ |
| H A D | AssetsLibrary.h | 506 void AssetsLibrary::fill_borders_with_garbage(T &&tensor, D &&distribution, std::random_device::res… in fill_borders_with_garbage()
535 void AssetsLibrary::fill_boxes(T &&tensor, D &&distribution, std::random_device::result_type seed_o… in fill_boxes()
571 void AssetsLibrary::fill(std::vector<T> &vec, D &&distribution, std::random_device::result_type see… in fill()
622 void AssetsLibrary::fill(T &&tensor, D &&distribution, std::random_device::result_type seed_offset)… in fill()
658 void AssetsLibrary::fill(RawTensor &raw, D &&distribution, std::random_device::result_type seed_off… in fill()
|
| /aosp_15_r20/external/libcxx/test/std/numerics/rand/rand.dis/rand.dist.pois/rand.dist.pois.poisson/ |
| H A D | eval.pass.cpp | 39 std::poisson_distribution<std::int16_t> distribution(32710.9); in test_bad_ranges() local
46 std::poisson_distribution<std::int16_t> distribution(std::numeric_limits<std::int16_t>::max()); in test_bad_ranges() local
77 std::poisson_distribution<std::int16_t> distribution(0); in test_bad_ranges() local
85 std::poisson_distribution<std::int16_t> distribution(-100); in test_bad_ranges() local
|
| /aosp_15_r20/external/cronet/third_party/libc++/src/test/std/numerics/rand/rand.dist/rand.dist.pois/rand.dist.pois.poisson/ |
| H A D | eval.pass.cpp | 40 std::poisson_distribution<std::int16_t> distribution(32710.9); in test_bad_ranges() local
47 std::poisson_distribution<std::int16_t> distribution(std::numeric_limits<std::int16_t>::max()); in test_bad_ranges() local
78 std::poisson_distribution<std::int16_t> distribution(0); in test_bad_ranges() local
86 std::poisson_distribution<std::int16_t> distribution(-100); in test_bad_ranges() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | reshape_test.cc | 724 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
744 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
765 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
791 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
873 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
902 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
931 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
960 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
990 std::uniform_real_distribution<float> distribution; in XLA_TEST_P() local
|