| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| 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
|
| /external/XNNPACK/tools/ |
| D | generate-raddstoreexpminusmax-test.py | 81 def generate_test_cases(ukernel, init_fn, elements_tile, isa): argument
|
| D | generate-vmulcaddc-test.py | 208 def generate_test_cases(ukernel, channel_tile, row_tile, init_fn, isa): argument
|
| D | generate-vlrelu-test.py | 157 def generate_test_cases(ukernel, init_fn, datatype, batch_tile, isa): argument
|
| D | generate-vunary-test.py | 205 def generate_test_cases(ukernel, op_type, init_fn, batch_tile, isa): argument
|
| D | generate-avgpool-test.py | 2395 def generate_test_cases(ukernel, init_fn, primary_tile, incremental_tile, argument
|
| D | generate-vcvt-test.py | 270 def generate_test_cases(ukernel, init_fn, input_datatype, output_datatype, argument
|
| D | generate-maxpool-test.py | 1186 def generate_test_cases(ukernel, init_fn, primary_tile, incremental_tile, argument
|
| D | generate-gavgpool-test.py | 510 def generate_test_cases(ukernel, init_fn, requantization_type, primary_tile, argument
|
| D | generate-spmm-test.py | 379 def generate_test_cases(ukernel, init_fn, mr, nr, k_block, is_pipelined, isa): argument
|
| D | generate-vbinary-test.py | 264 def generate_test_cases(ukernel, op_type, init_fn, activation_type, argument
|
| D | generate-dwconv2d-chw-test.py | 244 init_fn, padding, isa, height_tile, width_tile): argument
|
| D | generate-dwconv-test.py | 297 init_fn, requantization, is_pipelined, isa): argument
|
| D | generate-gemm-test.py | 875 def generate_test_cases(ukernel, mr, nr, kr, sr, xw, k_block, init_fn, argument
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_v1_in_v2.py | 59 def __init__(self, init_fn, asset_paths): argument
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | group_by_reducer_test.py | 175 def init_fn(_): function
|
| /external/tensorflow/tensorflow/c/experimental/grappler/ |
| D | grappler.cc | 152 auto init_fn = reinterpret_cast<TFInitGraphPluginFn>(dso_symbol); in InitGraphPlugin() local
156 tensorflow::Status InitGraphPlugin(TFInitGraphPluginFn init_fn) { in InitGraphPlugin()
|
| /external/tensorflow/tensorflow/c/experimental/pluggable_profiler/ |
| D | pluggable_profiler.cc | 152 Status InitPluginProfiler(TFInitProfilerFn init_fn) { in InitPluginProfiler()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | session_manager.py | 260 init_fn=None): argument
|
| D | monitored_session.py | 103 init_fn=None, argument
244 def init_fn(self): member in Scaffold
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | multi_device_iterator_test.py | 370 def init_fn(n): function
|
| D | iterator_test.py | 962 def init_fn(n): function
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | ops_test.py | 421 def init_fn(): function
|
| /external/tensorflow/tensorflow/python/checkpoint/ |
| D | checkpoint_management.py | 546 init_fn=None): argument
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor.cc | 748 auto init_fn = reinterpret_cast<SEInitPluginFn>(dso_symbol); in InitStreamExecutorPlugin() local
752 port::Status InitStreamExecutorPlugin(SEInitPluginFn init_fn, in InitStreamExecutorPlugin()
|