| /external/tensorflow/tensorflow/python/ops/ |
| D | variable_scope.py | 60 def __init__(self, full_shape, var_offset): argument
79 if not isinstance(var_offset, (list, tuple)):
82 type(var_offset).__name__)
84 if len(var_offset) != len(full_shape):
88 len(var_offset), len(full_shape)))
90 for offset, shape in zip(var_offset, full_shape):
94 "var_offset={}, full_shape={}".format(offset, shape, var_offset,
98 self._var_offset = var_offset
105 def var_offset(self): member in _PartitionInfo
128 return self.var_offset[single_slice_dim]
[all …]
|
| D | variables.py | 1268 var_offset=None, argument
1292 self.var_offset = [i for i in save_slice_info_def.var_offset]
1297 self.var_offset = var_offset
1305 "%d,%d" % (o, s) for o, s in zip(self.var_offset, self.var_shape))
1324 for i in self.var_offset:
1325 save_slice_info_def.var_offset.append(i)
2939 variable_list, key=lambda v: v._get_save_slice_info().var_offset)
|
| D | init_ops.py | 1586 initializer = array_ops.slice(initializer, partition_info.var_offset,
|
| /external/mesa3d/src/mapi/glapi/gen/ |
| D | glX_server_table.py | 314 var_offset = "%2u" % (len(var_table))
317 var_offset = "~0"
319 print ' /* [%3u] = %5u */ {%3u, %s},' % (index, opcode, fixed, var_offset)
|
| /external/mesa3d/src/compiler/glsl/ |
| D | lower_shared_reference.cpp | 46 struct var_offset { struct
104 list_for_each_entry(var_offset, var_entry, &var_offsets, node) { in get_shared_offset()
109 struct var_offset *new_entry = rzalloc(list_ctx, struct var_offset); in get_shared_offset()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-variable.-save-slice-info.pbtxt | 11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.experimental.dtensor.-d-variable.-save-slice-info.pbtxt | 11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
|
| D | tensorflow.-variable.-save-slice-info.pbtxt | 11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | sharded_variable.py | 713 var_offset = [0 for _ in range(dims)]
718 var_offset=copy.copy(var_offset),
723 var_offset[0] += int(v.shape[0])
|
| D | sharded_variable_test.py | 457 full_name='s', full_shape=[2], var_offset=[0], var_shape=[1]))
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | variable_scope_test.py | 1757 variable_scope._PartitionInfo(full_shape=None, var_offset=[0, 1])
1759 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset=None)
1761 variable_scope._PartitionInfo(full_shape="foo", var_offset=[0, 1])
1763 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset="foo")
1767 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset=[0])
1770 variable_scope._PartitionInfo(full_shape=[1, 1], var_offset=[0, 1])
1776 full_shape=[9, 3], var_offset=[4, 0])
1781 full_shape=[9, 3], var_offset=[0, 0])
1788 full_shape=[9, 3], var_offset=[4, 0])
1806 full_shape=[9, 3], var_offset=[0, 0])
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | warm_starting_util.py | 261 full_shape=slice_info.full_shape, var_offset=slice_info.var_offset)
|
| D | slot_creator.py | 102 slice_info.full_shape[:n], slice_info.var_offset[:n],
|
| D | slot_creator_test.py | 156 self.assertAllEqual([i], si.var_offset)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variable.proto | 81 repeated int64 var_offset = 3; field
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_v2.py | 336 …part_name = name + '/' + str(v._get_save_slice_info().var_offset[0]) # pylint: disable=protected-…
|