Home
last modified time | relevance | path

Searched defs:get_weights (Results 1 – 19 of 19) sorted by relevance

/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEFullyConnectedLayer.h56 ITensor *get_weights() override in get_weights() function
/external/pytorch/torch/testing/_internal/distributed/_shard/
Dtest_common.py22 def get_weights(self): member in SimpleMegatronLM
/external/ComputeLibrary/arm_compute/runtime/CL/functions/
DCLConvertFullyConnectedWeights.h124 ICLTensor *get_weights() override in get_weights() function
/external/cronet/tot/third_party/rust/chromium_crates_io/vendor/simd-adler32-0.3.7/src/imp/
Davx2.rs156 unsafe fn get_weights() -> __m256i { in get_weights() function
Davx512.rs183 unsafe fn get_weights() -> __m512i { in get_weights() function
Dneon.rs182 unsafe fn get_weights() -> __m512i { in get_weights() function
/external/cronet/stable/third_party/rust/chromium_crates_io/vendor/simd-adler32-0.3.7/src/imp/
Davx2.rs156 unsafe fn get_weights() -> __m256i { in get_weights() function
Davx512.rs183 unsafe fn get_weights() -> __m512i { in get_weights() function
Dneon.rs182 unsafe fn get_weights() -> __m512i { in get_weights() function
/external/rust/android-crates-io/crates/simd-adler32/src/imp/
Davx2.rs156 unsafe fn get_weights() -> __m256i { in get_weights() function
Dneon.rs182 unsafe fn get_weights() -> __m512i { in get_weights() function
Davx512.rs183 unsafe fn get_weights() -> __m512i { in get_weights() function
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py140 def get_weights(self): member in Optimizer
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer.py752 def get_weights(self): member in LossScaleOptimizer
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2.py1083 def get_weights(self): member in OptimizerV2
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_v1.py1306 def get_weights(self): member in Layer
Dtraining.py2072 def get_weights(self): member in Model
Dtraining_v1.py168 def get_weights(self): member in Model
Dbase_layer.py1808 def get_weights(self): member in Layer