Home
last modified time | relevance | path

Searched refs:var_offset (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py68 def __init__(self, full_shape, var_offset): argument
88 if not isinstance(var_offset, collections_abc.Sequence) or isinstance(
89 var_offset, six.string_types):
92 type(var_offset).__name__)
94 if len(var_offset) != len(full_shape):
98 len(var_offset), len(full_shape)))
100 for offset, shape in zip(var_offset, full_shape):
104 "var_offset={}, full_shape={}".format(offset, shape, var_offset,
108 self._var_offset = var_offset
115 def var_offset(self): member in _PartitionInfo
[all …]
Dvariables.py1276 var_offset=None, argument
1300 self.var_offset = [i for i in save_slice_info_def.var_offset]
1305 self.var_offset = var_offset
1313 "%d,%d" % (o, s) for o, s in zip(self.var_offset, self.var_shape))
1332 for i in self.var_offset:
1333 save_slice_info_def.var_offset.append(i)
2945 variable_list, key=lambda v: v._get_save_slice_info().var_offset)
Dinit_ops.py1199 initializer = array_ops.slice(initializer, partition_info.var_offset,
/external/mesa3d/src/mapi/glapi/gen/
DglX_server_table.py314 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/
Dlower_shared_reference.cpp46 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/
Dtensorflow.-variable.-save-slice-info.pbtxt11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-variable.-save-slice-info.pbtxt11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py543 var_offset = [0 for _ in range(dims)]
548 var_offset=copy.copy(var_offset),
553 var_offset[0] += int(v.shape[0])
Dsharded_variable_test.py338 full_name='s', full_shape=[2], var_offset=[0], var_shape=[1]))
/external/tensorflow/tensorflow/python/training/
Dwarm_starting_util.py267 full_shape=slice_info.full_shape, var_offset=slice_info.var_offset)
Dslot_creator.py106 slice_info.full_shape[:n], slice_info.var_offset[:n],
Dslot_creator_test.py159 self.assertAllEqual([i], si.var_offset)
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py1760 variable_scope._PartitionInfo(full_shape=None, var_offset=[0, 1])
1762 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset=None)
1764 variable_scope._PartitionInfo(full_shape="foo", var_offset=[0, 1])
1766 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset="foo")
1770 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset=[0])
1773 variable_scope._PartitionInfo(full_shape=[1, 1], var_offset=[0, 1])
1779 full_shape=[9, 3], var_offset=[4, 0])
1784 full_shape=[9, 3], var_offset=[0, 0])
1791 full_shape=[9, 3], var_offset=[4, 0])
1809 full_shape=[9, 3], var_offset=[0, 0])
[all …]
/external/tensorflow/tensorflow/core/framework/
Dvariable.proto81 repeated int64 var_offset = 3; field
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2.py316 …part_name = name + '/' + str(v._get_save_slice_info().var_offset[0]) # pylint: disable=protected-…