| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_init.py | 14 def _run_init_op(self, init_op, *args, **kwargs): argument
38 def _run_init_op(self, init_op, dist_init_op, eq_op, *args, **kwargs): argument
|
| H A D | test_random_ops.py | 25 def _run_init_op(self, init_op, *args, **kwargs): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/training/ |
| H A D | supervisor.py | 210 init_op=USE_DEFAULT, argument
414 def _init_init_op(self, init_op=USE_DEFAULT, init_feed_dict=None): argument
529 def init_op(self): member in Supervisor
|
| H A D | monitored_session.py | 101 init_op=None, argument
248 def init_op(self): member in Scaffold
|
| H A D | session_manager.py | 252 init_op=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
| H A D | builder_impl.py | 236 init_op=None, argument
309 init_op=None, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/ |
| H A D | checkpoint_test_base.py | 528 init_op, argument
599 def _initialize(self, init_op, sess): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | freeze_saved_model_assets.cc | 97 for (auto init_op : init_table_from_text_file_ops_to_erase) { in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_global_init.cc | 99 auto init_op = ops::ConfigureDistributedTPU( in ConstructDistributedInitializationGraph() local
|
| /aosp_15_r20/external/libmpeg2/fuzzer/ |
| H A D | mpeg2_dec_fuzzer.cpp | 177 impeg2d_init_op_t init_op; in createCodec() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| H A D | tf_jitrt_tile_reduction.cc | 179 auto init_op = fill_op.output().getDefiningOp<InitTensorOp>(); in matchAndRewrite() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | sparsify_gather.cc | 451 NodeDef* init_op = nullptr; in SparsifyGatherInternal() local
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ |
| H A D | _api.py | 910 init_op, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/ |
| H A D | benchmark_model.cc | 57 for (const string& init_op : init_ops) { in InitializeVariables() local
|
| /aosp_15_r20/external/federated-compute/fcp/protos/ |
| H A D | plan.proto | 1351 string init_op = 1; field
|
| /aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-cc-proto-lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/fcp/protos/ |
| D | plan.pb.h | 19332 inline const std::string& CrossRoundAggregationExecution::init_op() const { in init_op() function
19363 inline void CrossRoundAggregationExecution::set_allocated_init_op(std::string* init_op) { in set_allocated_init_op()
|