Searched refs:parallelism_parameters (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | model.cc | 223 Node::ModelParameters* parallelism_parameters, in CollectParameters() argument
237 parallelism_parameters->push_back(pair); in CollectParameters()
2254 const Model::ModelParameters& parallelism_parameters, in ShouldStop() argument
2262 for (auto& pair : parallelism_parameters) { in ShouldStop()
2350 Model::ModelParameters parallelism_parameters, buffer_size_parameters; in OptimizeGradientDescent() local
2351 CollectParameters(snapshot, parameters, ¶llelism_parameters, in OptimizeGradientDescent()
2377 parallelism_parameters, buffer_size_parameters, snapshot, in OptimizeGradientDescent()
|
| H A D | model.h | 943 const ModelParameters& parallelism_parameters,
|