Home
last modified time | relevance | path

Searched defs:var_device (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/hardware/google/gfxstream/host/magma/magma_dec/
Dmagma_dec.cpp68 magma_device_t var_device = Unpack<magma_device_t,uint64_t>(ptr + 8); in decode() local
81 magma_device_t var_device = Unpack<magma_device_t,uint64_t>(ptr + 8); in decode() local
110 magma_device_t var_device = Unpack<magma_device_t,uint64_t>(ptr + 8); in decode() local
148 magma_device_t var_device = Unpack<magma_device_t,uint64_t>(ptr + 8); in decode() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/optimizer_v2/
H A Dadam.py132 def _prepare_local(self, var_device, var_dtype, apply_state): argument
384 def _prepare_local(self, var_device, var_dtype, apply_state): argument
H A Dadamax.py110 def _prepare_local(self, var_device, var_dtype, apply_state): argument
H A Dftrl.py153 def _prepare_local(self, var_device, var_dtype, apply_state): argument
H A Dadadelta.py89 def _prepare_local(self, var_device, var_dtype, apply_state): argument
H A Dnadam.py106 def _prepare_local(self, var_device, var_dtype, apply_state): argument
H A Dgradient_descent.py124 def _prepare_local(self, var_device, var_dtype, apply_state): argument
H A Drmsprop.py160 def _prepare_local(self, var_device, var_dtype, apply_state): argument
H A Dadagrad.py91 def _prepare_local(self, var_device, var_dtype, apply_state): argument
H A Doptimizer_v2.py951 def _prepare_local(self, var_device, var_dtype, apply_state): argument
956 def _fallback_apply_state(self, var_device, var_dtype): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc2779 DeviceNameUtils::ParsedName var_device; in FindVariableInputs() local