Home
last modified time | relevance | path

Searched refs:init_variable (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/
H A Dall_reduce_test.py134 state_ops.init_variable(var, npt).op.run()
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dtriton_kernel_wrap.py798 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 Dstate_ops.py78 def init_variable(v, init, name="init"): function