Searched refs:init_variable (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | all_reduce_test.py | 134 state_ops.init_variable(var, npt).op.run()
|
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | triton_kernel_wrap.py | 798 def init_variable(self, variable, kernel, kernel_idx, grid): member in TritonHOPifier
1020 tracing_triton_hopifier_singleton.init_variable(self, kernel, kernel_idx, grid)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | state_ops.py | 78 def init_variable(v, init, name="init"): function
|