/external/tensorflow/tensorflow/python/ops/signal/ |
D | reconstruction_ops.py | 81 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/ |
D | variable_scope.py | 68 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 …]
|
D | init_ops.py | 416 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)
|
D | variables.py | 1275 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])))
|
D | init_ops_v2_test.py | 79 full_shape = (4, 2) 82 full_value = self.evaluate(init(full_shape, dtype=dtypes.float32)) 85 full_shape,
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | svd_op_test.py | 377 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,
|
D | variable_scope_test.py | 1760 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/ |
D | xla_sharding.py | 389 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/ |
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.-variable.-save-slice-info.pbtxt | 11 …argspec: "args=[\'self\', \'full_name\', \'full_shape\', \'var_offset\', \'var_shape\', \'save_sli…
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | cpu_instruction_fusion_test.cc | 553 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/ |
D | warm_starting_util.py | 267 full_shape=slice_info.full_shape, var_offset=slice_info.var_offset)
|
D | slot_creator.py | 106 slice_info.full_shape[:n], slice_info.var_offset[:n],
|
D | slot_creator_test.py | 158 self.assertAllEqual([2], si.full_shape)
|
/external/tensorflow/tensorflow/core/util/tensor_bundle/ |
D | tensor_bundle.cc | 998 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/ |
D | variable.proto | 79 repeated int64 full_shape = 2; field
|
/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
D | xla.py | 464 full_shape=op.inputs[0].shape.as_list())
|
/external/tensorflow/tensorflow/python/distribute/ |
D | sharded_variable.py | 547 full_shape=self.shape.as_list(),
|
D | sharded_variable_test.py | 338 full_name='s', full_shape=[2], var_offset=[0], var_shape=[1]))
|
/external/tensorflow/tensorflow/python/keras/feature_column/ |
D | dense_features_test.py | 748 partition_info.full_shape)
|
/external/tensorflow/tensorflow/python/feature_column/ |
D | feature_column_test.py | 4950 partition_info.full_shape) 5807 partition_info.full_shape)
|
D | feature_column_v2_test.py | 5734 partition_info.full_shape)
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | algebraic_simplifier_test.cc | 4633 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/ |
D | wrappers.go | 46 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/ |
D | RELEASE.md | 4416 * `uniform_unit_scaling_initializer()` no longer takes a `full_shape` arg,
|