Home
last modified time | relevance | path

Searched defs:partitioner (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dwarm_starting_util_test.py58 partitioner=None): argument
97 def _create_linear_model(self, feature_cols, partitioner): argument
564 partitioner = lambda shape, dtype: [1] * len(shape) function
593 partitioner = lambda shape, dtype: [1] * len(shape) function
627 partitioner = lambda shape, dtype: [1] * len(shape) function
663 partitioner = lambda shape, dtype: [1] * len(shape) function
708 partitioner = lambda shape, dtype: [1] * len(shape) function
751 partitioner = lambda shape, dtype: [1] * len(shape) function
822 partitioner = lambda shape, dtype: [1] * len(shape) function
H A Dsaver_test.py1075 def _save(partitioner=None): argument
1106 def _restore(partitioner=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dvariable_scope.py326 partitioner=None, argument
475 partitioner=None, argument
601 partitioner, argument
1148 partitioner=None, argument
1206 def partitioner(self): member in VariableScope
1247 def set_partitioner(self, partitioner): argument
1283 partitioner=None, argument
1354 partitioner=None, argument
1609 partitioner=None, argument
1786 partitioner=None, argument
[all …]
H A Dpartitioned_variables.py333 partitioner = lambda **unused_kwargs: slicing function
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
H A Dvariable_scope_shim.py164 partitioner=None, argument
307 partitioner=None, argument
H A Dbase.py334 partitioner=None, argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_spmd_partitioner.h34 spmd::SpmdPartitionerOptions options, spmd::SpmdPartitioner* partitioner) in GpuSpmdPartitioningVisitor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dstateful_rng_spmd_partitioner.h35 spmd::SpmdPartitionerOptions options, spmd::SpmdPartitioner* partitioner) in StatefulRngSpmdPartitioningVisitor()
H A Dspmd_partitioner.h331 SpmdPartitioner* partitioner; member
H A Dspmd_partitioner.cc1992 SpmdPartitionerOptions options, SpmdPartitioner* partitioner) in SpmdPartitioningVisitor()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/io_ops/
H A Dcheckpoint_ops_test.py326 def _test_loading_variable_with_max_rows(self, np_value, partitioner, argument
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dbase_preprocessing_layer.py372 partitioner=None, argument
H A Dbase_layer_utils.py63 partitioner=None): # pylint: disable=unused-argument argument
H A Dbase_layer_v1.py325 partitioner=None, argument
/aosp_15_r20/external/executorch/backends/qualcomm/utils/
H A Dutils.py436 partitioner, argument
/aosp_15_r20/external/pytorch/test/higher_order_ops/
H A Dtest_with_effects.py53 f, inps, partitioner=min_cut_rematerialization_partition, dynamic=False argument
/aosp_15_r20/external/executorch/exir/program/test/
H A Dtest_program.py569 def get_num_nondecomposed_ops(self, ep, partitioner): argument
/aosp_15_r20/external/executorch/exir/program/
H A D_program.py764 partitioner, argument
/aosp_15_r20/external/icu/tools/srcgen/currysrc/libs/
HDorg.eclipse.text_3.6.300.v20180430-1330.jarMETA-INF/MANIFEST.MF META-INF/ECLIPSE_.SF META-INF/ECLIPSE_ ...
HDorg.eclipse.jdt.core_3.14.0.v20180528-0519.jarMETA-INF/MANIFEST.MF META-INF/ECLIPSE_.SF META-INF/ECLIPSE_ ...
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_aotdispatch.py4004 f, inps, partitioner=min_cut_rematerialization_partition, dynamic=False argument
/aosp_15_r20/out/soong/.intermediates/prebuilts/tools/common/m2/dialer-dagger2-compiler-deps/linux_glibc_common/local-combined/
Ddialer-dagger2-compiler-deps.jarMETA-INF/ META-INF/MANIFEST.MF META-INF/eclipse.inf META ...
/aosp_15_r20/prebuilts/tools/common/m2/repository/com/google/dagger/dagger-compiler/2.7/
HDdagger-compiler-2.7.jarMETA-INF/ META-INF/MANIFEST.MF dagger/ dagger/internal/ dagger/ ...
/aosp_15_r20/external/guice/extensions/struts2/lib/
HDcore-3.1.1.jarMETA-INF/ org/ org/eclipse/ org/eclipse/jdt/ org/eclipse ...