| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_plugin_init.cc | 47 auto init_fn = in InitDeviceAndGraphModule() local
66 auto init_fn = in InitDeviceAndGraphModule() local
81 auto init_fn = reinterpret_cast<TFKernelInitFn>(dso_symbol); in InitKernelModule() local
92 auto init_fn = reinterpret_cast<profiler::TFInitProfilerFn>(dso_symbol); in InitProfilerModule() local
|
| /aosp_15_r20/external/pytorch/torch/jit/ |
| H A D | _recursive.py | 579 def init_fn(script_module): function
1013 def init_fn(script_module): function
1048 def init_fn(script_module): function
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_meta.py | 153 def _test_simple_model_with_meta_device(self, meta_module_fn, init_fn=None): argument
238 self, auto_wrap, meta_module_fn, init_fn=None argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/grappler/ |
| H A D | grappler.cc | 152 auto init_fn = reinterpret_cast<TFInitGraphPluginFn>(dso_symbol); in InitGraphPlugin() local
156 tensorflow::Status InitGraphPlugin(TFInitGraphPluginFn init_fn) { in InitGraphPlugin()
|
| /aosp_15_r20/external/XNNPACK/tools/ |
| H A D | generate-raddstoreexpminusmax-test.py | 81 def generate_test_cases(ukernel, init_fn, elements_tile, isa): argument
|
| H A D | generate-vmulcaddc-test.py | 208 def generate_test_cases(ukernel, channel_tile, row_tile, init_fn, isa): argument
|
| H A D | generate-vlrelu-test.py | 157 def generate_test_cases(ukernel, init_fn, datatype, batch_tile, isa): argument
|
| H A D | generate-vunary-test.py | 205 def generate_test_cases(ukernel, op_type, init_fn, batch_tile, isa): argument
|
| H A D | generate-vcvt-test.py | 270 def generate_test_cases(ukernel, init_fn, input_datatype, output_datatype, argument
|
| H A D | generate-maxpool-test.py | 1186 def generate_test_cases(ukernel, init_fn, primary_tile, incremental_tile, argument
|
| H A D | generate-avgpool-test.py | 2395 def generate_test_cases(ukernel, init_fn, primary_tile, incremental_tile, argument
|
| H A D | generate-gavgpool-test.py | 510 def generate_test_cases(ukernel, init_fn, requantization_type, primary_tile, argument
|
| H A D | generate-spmm-test.py | 379 def generate_test_cases(ukernel, init_fn, mr, nr, k_block, is_pipelined, isa): argument
|
| H A D | generate-vbinary-test.py | 264 def generate_test_cases(ukernel, op_type, init_fn, activation_type, argument
|
| H A D | generate-dwconv2d-chw-test.py | 244 init_fn, padding, isa, height_tile, width_tile): argument
|
| H A D | generate-dwconv-test.py | 297 init_fn, requantization, is_pipelined, isa): argument
|
| H A D | generate-gemm-test.py | 875 def generate_test_cases(ukernel, mr, nr, kr, sr, xw, k_block, init_fn, argument
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/ |
| H A D | OperatorRegistry.h | 68 explicit OperatorRegisterInit(InitFn* init_fn) { in OperatorRegisterInit()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/api/ |
| H A D | ShaderRegistry.h | 73 ShaderRegisterInit(InitFn* init_fn) { in ShaderRegisterInit()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/api/ |
| H A D | ShaderRegistry.h | 77 ShaderRegisterInit(InitFn* init_fn) { in ShaderRegisterInit()
|
| /aosp_15_r20/external/libdav1d/src/loongarch/ |
| H A D | ipred.h | 40 #define init_fn(type0, type1, name, suffix) \ macro
|
| /aosp_15_r20/external/libdav1d/src/x86/ |
| H A D | ipred.h | 35 #define init_fn(type0, type1, name, suffix) \ macro
|
| /aosp_15_r20/external/pytorch/functorch/examples/ensembling/ |
| H A D | parallel_train.py | 116 def init_fn(num_models): function
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_custom_class.cpp | 18 Function* init_fn = instance.type()->findMethod("__init__"); in __call__() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
| H A D | load_v1_in_v2.py | 59 def __init__(self, init_fn, asset_paths): argument
|