Searched defs:local_step (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | data_flow_ops.py | 1359 def apply_grad(self, grad, local_step=0, name=None): argument
1448 def apply_indexed_slices_grad(self, grad, local_step=0, name=None): argument
1476 local_step=0, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | typed_conditional_accumulator_base.h | 52 void TryApplyGrad(int64_t local_step, OpKernelContext* ctx) override { in TryApplyGrad()
|
| /aosp_15_r20/external/pytorch/torch/distributed/checkpoint/ |
| H A D | state_dict_loader.py | 205 def local_step(): function
|
| H A D | state_dict_saver.py | 285 def local_step(): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | moving_averages.py | 267 def update_fn(v, value, biased_var, local_step): argument
|