| /external/tensorflow/tensorflow/python/framework/ |
| H A D | tensor_shape_test.py | 22 from tensorflow.python.framework import tensor_shape namespace
30 dim = tensor_shape.Dimension(12)
33 self.assertEqual(dim, tensor_shape.Dimension(12))
34 self.assertEqual(tensor_shape.Dimension(15),
35 dim + tensor_shape.Dimension(3))
36 self.assertEqual(tensor_shape.Dimension(15), dim + 3)
37 self.assertEqual(tensor_shape.Dimension(24),
38 dim * tensor_shape.Dimension(2))
39 self.assertEqual(tensor_shape.Dimension(24), dim * 2)
41 tensor_shape
[all...] |
| H A D | common_shapes_test.py | 24 from tensorflow.python.framework import tensor_shape namespace
66 s1 = tensor_shape.vector(5)
67 s2 = tensor_shape.vector(7)
69 unknown = tensor_shape.unknown_shape()
70 scalar = tensor_shape.scalar()
71 expanded_scalar = tensor_shape.TensorShape([1])
92 unknown = tensor_shape.unknown_shape()
93 shape_0 = tensor_shape.scalar()
94 shape_1 = tensor_shape.vector(1)
95 shape_4 = tensor_shape
[all...] |
| H A D | tensor_shape_div_test.py | 23 from tensorflow.python.framework import tensor_shape namespace
33 values = [tensor_shape.Dimension(x) for x in (3, 7, 11, None)]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | shape_util.h | 22 #include "tensorflow/core/framework/tensor_shape.h"
30 TensorShape* tensor_shape);
35 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape,
|
| H A D | shape_util.cc | 28 TensorShape* tensor_shape) {
34 *tensor_shape = TensorShape();
36 tensor_shape->AddDim(shape.dimensions(i));
42 Status TensorShapeToXLAShape(DataType dtype, const TensorShape& tensor_shape, argument
44 int rank = tensor_shape.dims();
48 dimensions[d] = tensor_shape.dim_size(d);
27 XLAShapeToTensorShape(const xla::Shape& shape, TensorShape* tensor_shape) argument
|
| /external/tensorflow/tensorflow/python/keras/_impl/keras/ |
| H A D | initializers_test.py | 52 tensor_shape = (9, 6, 7)
57 tensor_shape,
61 tensor_shape = (8, 12, 99)
64 tensor_shape,
68 tensor_shape = (12, 99, 7)
73 tensor_shape,
77 tensor_shape = (5, 6, 4)
79 self._runner(keras.initializers.Constant(2), tensor_shape,
83 tensor_shape = (5, 6, 4, 2)
85 fan_in, _ = init_ops._compute_fans(tensor_shape)
[all...] |
| /external/tensorflow/tensorflow/python/data/util/ |
| H A D | sparse_test.py | 27 from tensorflow.python.framework import tensor_shape namespace
89 "types": tensor_shape.scalar(),
91 "expected": tensor_shape.scalar()
94 "types": tensor_shape.scalar(),
96 "expected": tensor_shape.unknown_shape()
99 "types": (tensor_shape.scalar()),
101 "expected": (tensor_shape.scalar())
104 "types": (tensor_shape.scalar()),
106 "expected": (tensor_shape.unknown_shape())
109 "types": (tensor_shape
[all...] |
| /external/tensorflow/tensorflow/python/grappler/ |
| H A D | datasets_test.py | 28 from tensorflow.python.framework import tensor_shape namespace
39 'shape': tensor_shape.TensorShape([])
42 'shape': tensor_shape.TensorShape([3])
45 'shape': tensor_shape.TensorShape([1, 3])
64 'shape': tensor_shape.TensorShape([])
67 'shape': tensor_shape.TensorShape([3])
70 'shape': tensor_shape.TensorShape([1, 3])
89 'shape': tensor_shape.TensorShape([])
92 'shape': tensor_shape.TensorShape([3])
95 'shape': tensor_shape
[all...] |
| /external/tensorflow/tensorflow/contrib/mpi_collectives/ |
| H A D | mpi_message.proto | 20 import "tensorflow/core/framework/tensor_shape.proto";
39 TensorShapeProto tensor_shape = 5;
|
| /external/tensorflow/tensorflow/contrib/tensorboard/plugins/projector/ |
| H A D | projector_config.proto | 32 repeated uint32 tensor_shape = 4;
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| H A D | accumulate_n_v2.py | 24 from tensorflow.python.framework import tensor_shape namespace
80 shape = tensor_shape.as_shape(shape)
82 shape = tensor_shape.unknown_shape()
|
| H A D | script_ops.py | 26 from tensorflow.python.framework import tensor_shape namespace
137 output_types, tensor_shape.as_shape, output_shapes)
|
| /external/tensorflow/tensorflow/contrib/gan/python/features/python/ |
| H A D | conditioning_utils_impl.py | 41 tensor_shape = array_ops.shape(tensor)
42 static_tensor_shape = tensor_util.constant_value(tensor_shape)
44 tensor_shape)
|
| /external/tensorflow/tensorflow/python/eager/ |
| H A D | graph_only_ops.py | 25 from tensorflow.python.framework import tensor_shape namespace
46 shape = tensor_shape.TensorShape(shape)
47 assert isinstance(shape, tensor_shape.TensorShape)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | load_and_remap_matrix_op.cc | 138 TensorShape tensor_shape; variable
140 old_tensor_name, &tensor_type, &tensor_shape));
148 context, tensor_shape.dims() == 2,
151 tensor_shape.DebugString(), " of invalid rank ",
152 tensor_shape.dims(), " instead of expected shape of rank 2.")));
158 OP_REQUIRES(context, num_cols_ == tensor_shape.dim_size(1),
161 tensor_shape.DebugString(),
163 tensor_shape.dim_size(1),
170 TensorSlice slice(tensor_shape.dims());
207 tensor_slice.SliceTensorShape(tensor_shape,
[all...] |
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| H A D | tpu_sharding.py | 24 from tensorflow.python.framework import tensor_shape namespace
51 self._shard_dimension = tensor_shape.as_dimension(
120 self._shard_dimension = tensor_shape.as_dimension(shard_dimension)
169 shape = tensor_shape.as_shape(shape)
189 return tensor_shape.as_shape(dims)
205 shape = tensor_shape.as_shape(shape)
217 return tensor_shape.as_shape(dims)
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| H A D | stats_accumulator_ops_test.py | 23 from tensorflow.python.framework import tensor_shape namespace
35 gradient_shape=tensor_shape.scalar(),
36 hessian_shape=tensor_shape.scalar())
63 gradient_shape=tensor_shape.scalar(),
64 hessian_shape=tensor_shape.scalar())
92 gradient_shape=tensor_shape.scalar(),
93 hessian_shape=tensor_shape.scalar())
124 gradient_shape=tensor_shape.scalar(),
125 hessian_shape=tensor_shape.scalar())
165 gradient_shape=tensor_shape
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | sendrecv_ops.cc | 68 TensorShape tensor_shape; local
70 OP_REQUIRES_OK(ctx, ctx->GetAttr("shape", &tensor_shape));
72 OP_REQUIRES_OK(ctx, TensorShapeToXLAShape(dtype, tensor_shape, &shape_));
|
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/ |
| H A D | conditional_distribution_test.py | 24 from tensorflow.python.framework import tensor_shape namespace
35 self._static_batch_shape = tensor_shape.TensorShape(batch_shape)
36 self._static_event_shape = tensor_shape.TensorShape(event_shape)
|
| /external/tensorflow/tensorflow/python/estimator/export/ |
| H A D | export_output_test.py | 85 tensor_shape=shape))
90 tensor_shape=shape))
115 tensor_shape=shape))
120 tensor_shape=shape))
151 tensor_shape=shape))
156 tensor_shape=shape))
161 tensor_shape=shape))
191 tensor_shape=shape))
196 tensor_shape=shape))
|
| /external/tensorflow/tensorflow/contrib/kafka/python/ops/ |
| H A D | kafka_dataset_ops.py | 24 from tensorflow.python.framework import tensor_shape namespace
70 return tensor_shape.scalar()
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| H A D | input_fn_utils.py | 32 from tensorflow.python.framework import tensor_shape namespace
108 shape = tensor_shape.TensorShape(shape_list)
|
| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| H A D | alpha_dropout.py | 22 from tensorflow.python.framework import tensor_shape namespace
64 keep_prob.get_shape().assert_is_compatible_with(tensor_shape.scalar())
|
| /external/tensorflow/tensorflow/contrib/signal/python/kernel_tests/ |
| H A D | mfcc_ops_test.py | 23 from tensorflow.python.framework import tensor_shape namespace
58 random_ops.random_normal((2, 3, 5)), tensor_shape.TensorShape(None))
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| H A D | device_setter.py | 25 from tensorflow.python.framework import tensor_shape namespace
134 shape = tensor_shape.TensorShape(op.get_attr("shape"))
|