Home
last modified time | relevance | path

Searched defs:num_steps (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/utils/
Dstep_intersection.h33 uint32 num_steps; // aligned for how many steps. member
Dstep_intersection.cc154 uint32 num_steps) { in StringSrcToDstIndexMap()
230 uint32 num_steps = min_end_chief_idx - max_begin_chief_idx; in StepIntersection() local
Dstep_intersection_test.cc33 PerCoreStepInfo CreateOneTestStep(uint32 host_id, uint32 num_steps, in CreateOneTestStep()
/external/tensorflow/tensorflow/python/training/
Devaluation_test.py71 def _train_model(self, checkpoint_dir, num_steps): argument
Dbasic_session_run_hooks.py331 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
/external/tensorflow/tensorflow/core/grappler/clusters/
Dcluster.cc40 void Cluster::SetNumWarmupSteps(int num_steps) { in SetNumWarmupSteps()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dround_weights.cc35 int32_t num_steps; in RoundWeights() local
/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
Dc_api_multi_client_function_test.cc175 uint num_steps = 1; member
/external/tensorflow/tensorflow/core/tfrt/saved_model/
Dsaved_model_testutil.cc232 SavedModel* saved_model, const bool profile, const int32_t num_steps) { in ProcessPredictRequestsAndMaybeProfile()
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dfault_tolerance_test.py109 def schedule_training_functions(self, num_steps): argument
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc236 const int num_steps = lstm_input_.size(); in VerifyGoldens() local
/external/libdrm/tests/exynos/
Dexynos_fimg2d_test.c348 static const unsigned int num_steps = in g2d_move_test() local
/external/tensorflow/tensorflow/core/kernels/
Dquantization_utils_test.cc145 int num_steps = 14419; in TestRequantizeManyInNewRange32To8Bit() local
/external/tensorflow/tensorflow/lite/kernels/
Dlstm_test.cc353 const int num_steps = lstm_input_.size(); in VerifyGoldens() local