Home
last modified time | relevance | path

Searched refs:full_shape (Results 1 – 25 of 25) sorted by relevance

/external/tensorflow/tensorflow/python/ops/signal/
Dreconstruction_ops.py81 def full_shape(inner_shape): function
94 output_shape = full_shape([output_length])
128 shape = full_shape([frames + segments, segments, frame_step])
143 shape = full_shape([(frames + segments) * segments, frame_step])
154 shape = full_shape([segments, (frames + segments - 1), frame_step])
161 shape = full_shape([(frames + segments - 1) * frame_step])
/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py68 def __init__(self, full_shape, var_offset): argument
82 if not isinstance(full_shape, collections_abc.Sequence) or isinstance(
83 full_shape, six.string_types):
86 type(full_shape).__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):
105 full_shape))
107 self._full_shape = full_shape
111 def full_shape(self): member in _PartitionInfo
[all …]
Dinit_ops.py416 scale_shape = partition_info.full_shape
500 scale_shape = partition_info.full_shape
1189 full_shape = shape if partition_info is None else partition_info.full_shape
1190 if len(full_shape) != 2:
1195 if isinstance(full_shape, tensor_shape.TensorShape):
1196 full_shape = full_shape.as_list()
1197 initializer = linalg_ops_impl.eye(*full_shape, dtype=dtype)
Dvariables.py1275 full_shape=None, argument
1299 self.full_shape = [i for i in save_slice_info_def.full_shape]
1304 self.full_shape = full_shape
1311 full_shape_str = " ".join("%d" % d for d in self.full_shape) + " "
1330 for i in self.full_shape:
1331 save_slice_info_def.full_shape.append(i)
2940 if v._get_save_slice_info().full_shape != shape:
2943 (shape, str([v._get_save_slice_info().full_shape])))
Dinit_ops_v2_test.py79 full_shape = (4, 2)
82 full_value = self.evaluate(init(full_shape, dtype=dtypes.float32))
85 full_shape,
/external/tensorflow/tensorflow/python/kernel_tests/
Dsvd_op_test.py377 full_shape = batch_dims + (rows, cols) variable
380 dtype.__name__, "_".join(map(str, full_shape)),
384 _GetSvdOpTest(dtype, full_shape, use_static_shape,
396 full_shape = batch_dims + mat_shape variable
398 map(str, full_shape)), compute_uv, full_matrices)
401 _GetSvdGradOpTest(dtype, full_shape, compute_uv, full_matrices))
406 _GetSvdGradGradOpTest(dtype, full_shape, compute_uv,
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/compiler/xla/experimental/xla_sharding/
Dxla_sharding.py389 def manual_to_auto_spmd_partition(tensor, manual_sharding, full_shape): argument
406 tensor, manual_sharding=manual_sharding, full_shape=full_shape)
/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/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc553 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local
569 HloInstruction::CreateParameter(0, full_shape, "slice_from")), in TEST_F()
574 full_shape, in TEST_F()
576 HloInstruction::CreateParameter(4, full_shape, "to_update")), in TEST_F()
592 Shape full_shape = ShapeUtil::MakeShape(F32, {4, 100, 10, 100, 50}); in TEST_F() local
618 full_shape, in TEST_F()
620 HloInstruction::CreateParameter(4, full_shape, "param4")), in TEST_F()
/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.py158 self.assertAllEqual([2], si.full_shape)
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc998 const TensorShape full_shape(TensorShape(full_tensor_entry.shape())); in GetSliceValue() local
1011 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), in GetSliceValue()
1013 /* full slice */ TensorSlice(full_shape.dims()), &tensor_slices_)); in GetSliceValue()
1017 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), in GetSliceValue()
1079 full_shape, stored_slice, slice_spec, \ in GetSliceValue()
/external/tensorflow/tensorflow/core/framework/
Dvariable.proto79 repeated int64 full_shape = 2; field
/external/tensorflow/tensorflow/compiler/tf2xla/python/
Dxla.py464 full_shape=op.inputs[0].shape.as_list())
/external/tensorflow/tensorflow/python/distribute/
Dsharded_variable.py547 full_shape=self.shape.as_list(),
Dsharded_variable_test.py338 full_name='s', full_shape=[2], var_offset=[0], var_shape=[1]))
/external/tensorflow/tensorflow/python/keras/feature_column/
Ddense_features_test.py748 partition_info.full_shape)
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py4950 partition_info.full_shape)
5807 partition_info.full_shape)
Dfeature_column_v2_test.py5734 partition_info.full_shape)
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier_test.cc4633 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local
4649 HloInstruction::CreateParameter(0, full_shape, "slice_from")), in TEST_F()
/external/tensorflow/tensorflow/go/op/
Dwrappers.go46 func XlaSpmdShardToFullShape(scope *Scope, input tf.Output, manual_sharding string, full_shape tf.S…
50 attrs := map[string]interface{}{"manual_sharding": manual_sharding, "full_shape": full_shape}
/external/tensorflow/
DRELEASE.md4416 * `uniform_unit_scaling_initializer()` no longer takes a `full_shape` arg,