| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util_test.py | 63 partitioner=None): argument
70 partitioner=partitioner)
72 if partitioner:
102 def _create_linear_model(self, feature_cols, partitioner): argument
104 with variable_scope.variable_scope("", partitioner=partitioner):
138 partitioner=lambda shape, dtype: [2, 1])
161 partitioner=lambda shape, dtype: [2, 1])
178 partitioner=lambda shape, dtype: [2, 1])
187 partitioner=lambda shape, dtype: [2, 1])
278 partitioner=lambda shape, dtype: [2, 1])
[all …]
|
| D | checkpoint_ops_test.py | 156 partitioner=partitioned_variables.fixed_size_partitioner(2))
189 partitioner=partitioned_variables.fixed_size_partitioner(2))
226 partitioner=partitioned_variables.fixed_size_partitioner(2))
261 partitioner=partitioned_variables.fixed_size_partitioner(2))
294 partitioner=partitioned_variables.fixed_size_partitioner(2))
340 partitioner=partitioned_variables.fixed_size_partitioner(2))
378 partitioner=partitioned_variables.fixed_size_partitioner(2))
|
| D | checkpoint_utils_test.py | 71 partitioner=partitioned_variables.min_max_variable_partitioner(
278 partitioner=partitioned_variables.min_max_variable_partitioner(
288 partitioner=partitioned_variables.min_max_variable_partitioner(
316 partitioner=partitioned_variables.min_max_variable_partitioner(
|
| D | saver_large_partitioned_variable_test.py | 51 partitioner=partitioned_variables.fixed_size_partitioner(4),
|
| D | slot_creator_test.py | 147 partitioner=partitioned_variables.fixed_size_partitioner(2))
169 partitioner=partitioned_variables.fixed_size_partitioner(2))
|
| D | localhost_cluster_performance_test.py | 101 partitioner=partitioned_variables.variable_axis_size_partitioner(
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 320 partitioner=None, argument
469 partitioner=None, argument
479 if partitioner is not None and not is_scalar:
480 if not callable(partitioner):
482 partitioner)
494 partitioner=partitioner,
503 if (reuse is True and partitioner is None
515 partitioner=None,
563 "partitioner": partitioner,
586 partitioner=partitioner,
[all …]
|
| D | partitioned_variables.py | 298 partitioner = lambda **unused_kwargs: slicing function
309 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | partitioned_variables_test.py | 44 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
45 with variable_scope.variable_scope("root", partitioner=partitioner):
55 partitioner = partitioned_variables.fixed_size_partitioner(4, axis=0)
56 with variable_scope.variable_scope("root", partitioner=partitioner):
63 partitioner = partitioned_variables.fixed_size_partitioner(5, axis=0)
65 "root", partitioner=partitioner, use_resource=True):
80 partitioner = partitioned_variables.variable_axis_size_partitioner(
83 with variable_scope.variable_scope("root", partitioner=partitioner):
181 "root", partitioner=partitioner_axis3_str):
203 partitioner = partitioned_variables.min_max_variable_partitioner(
[all …]
|
| D | checkpoint_ops_test.py | 304 def _test_loading_variable_with_max_rows(self, np_value, partitioner, argument
313 partitioner=partitioner)
381 partitioner=None,
390 partitioner=None,
402 partitioner=None,
411 partitioner=None,
419 partitioner=partitioned_variables.fixed_size_partitioner(3),
429 partitioner=partitioned_variables.fixed_size_partitioner(3),
442 partitioner=partitioned_variables.fixed_size_partitioner(3),
450 partitioner=partitioned_variables.fixed_size_partitioner(3),
|
| D | variable_scope_test.py | 1414 "scope0", partitioner=axis0_into2_partitioner):
1428 "scope0", partitioner=axis0_into2_partitioner):
1432 "scope0", partitioner=axis0_into3_partitioner, reuse=True):
1440 "scope0", partitioner=axis0_into1_partitioner, reuse=True):
1451 "scope0", partitioner=axis0_into2_partitioner):
1461 "scope0", partitioner=axis0_into2_partitioner):
1470 "scope0", partitioner=axis0_into2_partitioner):
1479 "scope0", partitioner=axis0_into2_partitioner) as vs:
1480 self.assertEqual(axis0_into2_partitioner, vs.partitioner)
1482 self.assertEqual(axis0_into2_partitioner, vs1.partitioner)
[all …]
|
| /external/llvm-project/llvm/test/TableGen/GICombinerEmitter/ |
| D | match-tree.td | 90 // CHECK-DAG: Node[[N3:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule0}"]
91 // CHECK-DAG: Node[[N4:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule5}"]
92 // CHECK-DAG: Node[[N5:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule5}"]
95 // CHECK-DAG: Node[[N8:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule1}"]
96 // CHECK-DAG: Node[[N9:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule2}"]
97 // CHECK-DAG: Node[[N10:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule2}"]
100 // CHECK-DAG: Node[[N13:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule3,Rule4}",co…
101 // CHECK-DAG: Node[[N14:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule4}"]
102 // CHECK-DAG: Node[[N15:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule4}"]
105 // CHECK-DAG: Node[[N18:(0x)?[0-9a-fA-F]+]] [shape=record,label="{No partitioner|Rule6}"]
[all …]
|
| /external/tensorflow/tensorflow/python/keras/feature_column/ |
| D | dense_features.py | 73 partitioner=None, argument
97 partitioner=partitioner,
|
| D | base_feature_layer.py | 56 partitioner=None, argument
64 self._partitioner = partitioner
76 self.name, partitioner=self._partitioner):
|
| D | dense_features_test.py | 139 partitioner=partitioned_variables.fixed_size_partitioner(2))
772 partitioner = None
774 partitioner = partitioned_variables.fixed_size_partitioner(2, axis=0)
775 with variable_scope.variable_scope('vars', partitioner=partitioner):
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable-scope.pbtxt | 34 name: "partitioner"
51 …e\', \'name\', \'initializer\', \'regularizer\', \'caching_device\', \'partitioner\', \'custom_get…
59 …er\', \'reuse\', \'trainable\', \'collections\', \'caching_device\', \'partitioner\', \'validate_s…
91 argspec: "args=[\'self\', \'partitioner\'], varargs=None, keywords=None, defaults=None"
|
| D | tensorflow.variable_scope.pbtxt | 7 …', \'values\', \'initializer\', \'regularizer\', \'caching_device\', \'partitioner\', \'custom_get…
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable_test.py | 72 partitioner = sharded_variable.FixedShardsPartitioner(num_shards=2)
73 got = partitioner(tensor_shape.TensorShape([10, 3]), dtypes.float32)
77 partitioner = sharded_variable.MinSizePartitioner(
79 got = partitioner(tensor_shape.TensorShape([6, 1]), dtypes.float32)
82 partitioner = sharded_variable.MinSizePartitioner(
84 got = partitioner(tensor_shape.TensorShape([6, 1]), dtypes.float32)
88 partitioner = sharded_variable.MaxSizePartitioner(max_shard_bytes=4)
89 got = partitioner(tensor_shape.TensorShape([6, 1]), dtypes.float32)
92 partitioner = sharded_variable.MaxSizePartitioner(
94 got = partitioner(tensor_shape.TensorShape([6, 1]), dtypes.float32)
[all …]
|
| D | parameter_server_strategy_test.py | 226 partitioner = partitioned_variables.fixed_size_partitioner(num_shards)
236 partitioner=partitioner)
249 partitioner=partitioner)
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | base.py | 333 partitioner=None, argument
397 partitioner=partitioner,
461 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| D | rnn_cell.py | 322 def add_variable_wrapped(name, shape, initializer, index, partitioner): argument
324 name, shape=shape, initializer=initializer, partitioner=partitioner)
393 partitioner=maybe_proj_partitioner)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_preprocessing_layer.py | 327 partitioner=None, argument
353 partitioner=partitioner,
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | freeze_graph_test.py | 288 partitioner = partitioned_variables.fixed_size_partitioner(1)
290 with variable_scope.variable_scope("part", partitioner=partitioner):
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.h | 291 SpmdPartitioner* partitioner; member
406 SpmdPartitionerOptions options, SpmdPartitioner* partitioner);
491 state.partitioner = partitioner_; in MakePartitioningState()
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 442 partitioner = self._scope.partitioner
445 partitioner = None
516 if partitioner:
517 self._scope.set_partitioner(partitioner)
|