Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/ops/signal/
Dreconstruction_ops.py70 def full_shape(inner_shape): function
84 output_shape = full_shape([output_length])
118 shape = full_shape([frames + segments, segments, frame_step])
131 shape = full_shape([(frames + segments) * segments, frame_step])
142 shape = full_shape([segments, (frames + segments - 1), frame_step])
149 shape = full_shape([(frames + segments - 1) * frame_step])
/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py61 def __init__(self, full_shape, var_offset): argument
75 if not isinstance(full_shape, collections_lib.Sequence) or isinstance(
76 full_shape, six.string_types):
79 type(full_shape).__name__)
87 if len(var_offset) != len(full_shape):
91 len(var_offset), len(full_shape)))
93 for i in xrange(len(full_shape)):
95 shape = full_shape[i]
100 full_shape))
102 self._full_shape = full_shape
[all …]
Dinit_ops_v2.py417 scale_shape = partition_info.full_shape
540 full_shape = shape if partition_info is None else partition_info.full_shape
541 if len(full_shape) != 2:
544 initializer = linalg_ops_impl.eye(*full_shape, dtype=dtype)
Dinit_ops.py441 scale_shape = partition_info.full_shape
526 scale_shape = partition_info.full_shape
1206 full_shape = shape if partition_info is None else partition_info.full_shape
1207 if len(full_shape) != 2:
1212 if isinstance(full_shape, tensor_shape.TensorShape):
1213 full_shape = full_shape.as_list()
1214 initializer = linalg_ops_impl.eye(*full_shape, dtype=dtype)
Dvariables.py1083 full_shape=None, argument
1108 self.full_shape = [i for i in save_slice_info_def.full_shape]
1113 self.full_shape = full_shape
1120 full_shape_str = " ".join(["%d" % d for d in self.full_shape]) + " "
1141 for i in self.full_shape:
1142 save_slice_info_def.full_shape.append(i)
2510 if v._get_save_slice_info().full_shape != shape:
2514 % (shape, str([v._get_save_slice_info().full_shape])))
/external/tensorflow/tensorflow/contrib/distributions/python/ops/
Dsample_stats.py521 full_shape = front_shape + end_shape
525 full_shape = array_ops.concat([front_shape, end_shape], axis=0)
526 return array_ops.reshape(x_permed, shape=full_shape)
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dcheckpoint_ops_test.py143 full_shape=[5, 1], var_offset=[0, 0]),
155 full_shape=[5, 1], var_offset=[3, 0]),
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc551 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local
567 HloInstruction::CreateParameter(0, full_shape, "slice_from")), in TEST_F()
572 full_shape, in TEST_F()
574 HloInstruction::CreateParameter(4, full_shape, "to_update")), in TEST_F()
590 Shape full_shape = ShapeUtil::MakeShape(F32, {4, 100, 10, 100, 50}); in TEST_F() local
616 full_shape, in TEST_F()
618 HloInstruction::CreateParameter(4, full_shape, "param4")), in TEST_F()
/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/training/
Dwarm_starting_util.py258 full_shape=slice_info.full_shape,
Dslot_creator.py90 slice_info.full_shape[:],
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py1715 variable_scope._PartitionInfo(full_shape=None, var_offset=[0, 1])
1717 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset=None)
1719 variable_scope._PartitionInfo(full_shape="foo", var_offset=[0, 1])
1721 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset="foo")
1725 variable_scope._PartitionInfo(full_shape=[0, 1], var_offset=[0])
1728 variable_scope._PartitionInfo(full_shape=[1, 1], var_offset=[0, 1])
1734 full_shape=[9, 3], var_offset=[4, 0])
1739 full_shape=[9, 3], var_offset=[0, 0])
1746 full_shape=[9, 3], var_offset=[4, 0])
1764 full_shape=[9, 3], var_offset=[0, 0])
[all …]
/external/tensorflow/tensorflow/core/framework/
Dvariable.proto38 repeated int64 full_shape = 2; field
/external/tensorflow/tensorflow/core/util/tensor_bundle/
Dtensor_bundle.cc938 const TensorShape full_shape(TensorShape(full_tensor_entry.shape())); in GetSliceValue() local
951 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), in GetSliceValue()
953 /* full slice */ TensorSlice(full_shape.dims()), &tensor_slices_)); in GetSliceValue()
957 full_tensor_key_string, full_shape, full_tensor_entry.dtype(), in GetSliceValue()
1019 full_shape, stored_slice, slice_spec, \ in GetSliceValue()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc1260 Shape full_shape = ShapeUtil::MakeShape(F32, {16, 32}); in TEST_F() local
1265 HloInstruction::CreateParameter(0, full_shape, "full")); in TEST_F()
1269 HloInstruction::CreateBroadcast(full_shape, param1, {0})); in TEST_F()
1271 full_shape, HloOpcode::kAdd, param0, broadcast)); in TEST_F()
Dalgebraic_simplifier_test.cc3840 Shape full_shape = ShapeUtil::MakeShape(F32, {10, 100, 1000}); in TEST_F() local
3856 HloInstruction::CreateParameter(0, full_shape, "slice_from")), in TEST_F()
/external/tensorflow/
DRELEASE.md1743 * `uniform_unit_scaling_initializer()` no longer takes a `full_shape` arg,