Searched defs:call_fn (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | utils.py | 38 def use_wrapped_call(layer, call_fn, default_training_value=None, argument
199 def get_training_arg_index(call_fn): argument
|
| D | save_impl.py | 508 def _maybe_wrap_with_training_arg(self, call_fn, match_layer_training_arg): argument
547 def add_function(self, call_fn, name, match_layer_training_arg): argument
614 def __init__(self, call_collection, call_fn, name, input_signature): argument
|
| /external/pytorch/torch/csrc/autograd/functions/ |
| D | tensor.cpp | 78 const T& call_fn) { in apply_impl()
|
| /external/rust/android-crates-io/crates/pdl-compiler/src/backends/rust_no_allocation/ |
| D | computed_values.rs | 30 fn call_fn(&self) -> TokenStream { in call_fn() method
|
| /external/tensorflow/tensorflow/python/distribute/experimental/rpc/ |
| D | rpc_ops_test.py | 553 def call_fn(): function
|
| /external/federated-compute/fcp/secagg/server/ |
| D | secagg_scheduler_test.cc | 68 constexpr auto call_fn = [](const std::function<void()>& f) { f(); }; in __anon80561c260402() variable
|
| D | secagg_server_prng_running_state_test.cc | 76 constexpr auto call_fn = [](const std::function<void()>& f) { f(); }; in __anon9c7ef5f40202() variable
|
| /external/pytorch/torch/csrc/jit/tensorexpr/ |
| D | llvm_codegen.cpp | 2010 llvm::Value* call_fn = nullptr; in visit() local
2341 auto call_fn = callee.getCallee(); in visit() local
2513 auto call_fn = callee.getCallee(); in visit() local
2690 auto call_fn = callee.getCallee(); in visit() local
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | mkldnn_fusion.py | 80 def _unary_fusion_pattern(unary_fusion, call_fn, users, lowp_dtype): argument
|
| D | quantization.py | 112 def _unary_fusion_pattern(unary_fusion, call_fn, users, is_bf16): argument
|