| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | logging.py | 42 num_steps = 3 variable
44 num_steps = 2 variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/utils/ |
| H A D | step_intersection.cc | 154 uint32 num_steps) { in StringSrcToDstIndexMap()
230 uint32 num_steps = min_end_chief_idx - max_begin_chief_idx; in StepIntersection() local
|
| H A D | step_intersection.h | 33 uint32 num_steps; // aligned for how many steps. member
|
| H A D | step_intersection_test.cc | 33 PerCoreStepInfo CreateOneTestStep(uint32 host_id, uint32 num_steps, in CreateOneTestStep()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | evaluation_test.py | 71 def _train_model(self, checkpoint_dir, num_steps): argument
|
| H A D | basic_session_run_hooks.py | 331 def __init__(self, num_steps=None, last_step=None, steps_per_run=1): argument
404 def __init__(self, num_steps=None, last_step=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
| H A D | cluster.cc | 40 void Cluster::SetNumWarmupSteps(int num_steps) { in SetNumWarmupSteps()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | round_weights.cc | 35 int32_t num_steps; in RoundWeights() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | AffineGridGenerator.cpp | 18 static at::Tensor linspace_from_neg_one(const Tensor& grid, int64_t num_steps, in linspace_from_neg_one()
|
| H A D | RNN.cpp | 935 int64_t num_steps = input.batch_sizes.size(0); in operator ()() local
994 int64_t num_steps = input.batch_sizes.size(0); in operator ()() local
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/range/include/boost/range/ |
| D | irange.hpp | 226 const std::ptrdiff_t num_steps = (l - f) / sz + ((l - f) % sz ? 1 : 0); in irange() local
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_fine_tune.py | 135 def _assert_post_backward_reshard_count(step_idx, num_steps): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/ |
| H A D | c_api_multi_client_function_test.cc | 175 uint num_steps = 1; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| H A D | saved_model_testutil.cc | 232 SavedModel* saved_model, const bool profile, const int32_t num_steps) { in ProcessPredictRequestsAndMaybeProfile()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/ |
| H A D | fault_tolerance_test.py | 109 def schedule_training_functions(self, num_steps): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| H A D | lstm_full_test.cc | 236 const int num_steps = lstm_input_.size(); in VerifyGoldens() local
|
| /aosp_15_r20/external/libdrm/tests/exynos/ |
| H A D | exynos_fimg2d_test.c | 347 static const unsigned int num_steps = in g2d_move_test() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantization_utils_test.cc | 145 int num_steps = 14419; in TestRequantizeManyInNewRange32To8Bit() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | lstm_test.cc | 353 const int num_steps = lstm_input_.size(); in VerifyGoldens() local
|