Searched defs:var_device (Results 1 – 11 of 11) sorted by relevance
/aosp_15_r20/hardware/google/gfxstream/host/magma/magma_dec/ |
D | magma_dec.cpp | 68 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 D | adam.py | 132 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 D | adamax.py | 110 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
H A D | ftrl.py | 153 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
H A D | adadelta.py | 89 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
H A D | nadam.py | 106 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
H A D | gradient_descent.py | 124 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
H A D | rmsprop.py | 160 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
H A D | adagrad.py | 91 def _prepare_local(self, var_device, var_dtype, apply_state): argument
|
H A D | optimizer_v2.py | 951 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 D | distributed_tpu_rewrite_pass.cc | 2779 DeviceNameUtils::ParsedName var_device; in FindVariableInputs() local
|