Home
last modified time | relevance | path

Searched refs:LabeledTensor (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/
Dops_test.py60 self.original_lt = core.LabeledTensor(tensor, [a0, a1, a2, a3])
77 golden_lt = core.LabeledTensor(self.tensor[:, 1, :, :],
84 golden_lt = core.LabeledTensor(self.tensor[:, :2, :, :],
95 golden_lt = core.LabeledTensor(self.tensor[1:5, 1:, :, :],
102 golden_lt = core.LabeledTensor(self.tensor[:, :2, :, :],
109 golden_lt = core.LabeledTensor(self.tensor[:, :1, :, :],
115 golden_lt = core.LabeledTensor(self.tensor[:, :0, :, :],
121 golden_lt = core.LabeledTensor(self.tensor[1, 1, :, :], [self.a2, self.a3])
125 original_lt = core.LabeledTensor(constant_op.constant([5, 6]),
128 golden_lt = core.LabeledTensor(constant_op.constant(5), [])
[all …]
Dops.py39 @tc.returns(core.LabeledTensor)
40 @tc.accepts(core.LabeledTensor, ops.Tensor, core.Axis,
47 indexed = core.LabeledTensor(
52 @tc.returns(core.LabeledTensor)
145 @tc.returns(core.LabeledTensor)
208 return core.LabeledTensor(concat_tensor, concat_axes)
214 @tc.returns(core.LabeledTensor)
257 return core.LabeledTensor(pack_op, axes)
260 @tc.returns(tc.List(core.LabeledTensor))
293 return [core.LabeledTensor(t, axes) for t in unpack_ops]
[all …]
Dcore_test.py170 self.lt = core.LabeledTensor(tensor, [a0, a1, a2, a3])
183 alt_lt = core.LabeledTensor(self.lt.tensor, self.lt.axes)
187 alt_lt = core.LabeledTensor(self.lt.tensor, self.lt.axes.values())
192 expected = core.LabeledTensor(self.lt.tensor[:, :, :, 0],
197 expected = core.LabeledTensor(self.lt.tensor[1, :, :, 0],
202 expected = core.LabeledTensor(self.lt.tensor[1, 2, :, 0],
209 expected = core.LabeledTensor(lt_1d.tensor[3], [])
215 expected = core.LabeledTensor(self.lt.tensor[:3, :, :, :], axes)
226 actual = core.LabeledTensor(tensor, ['x'])
237 lt2 = core.LabeledTensor(self.lt.tensor, self.lt.axes)
[all …]
Dcore.py264 class LabeledTensor(object): class
477 if not isinstance(other, LabeledTensor):
497 tc.register_type_abbreviation(LabeledTensor, 'labeled_tensor.LabeledTensor')
501 @tc.accepts(LabeledTensor)
507 ops.register_tensor_conversion_function(LabeledTensor,
512 LabeledTensorLike = tc.Union(LabeledTensor, ops.Tensor, np.ndarray, Scalar)
516 @tc.returns(LabeledTensor)
539 if isinstance(value, LabeledTensor):
551 return LabeledTensor(tensor, axes)
594 @tc.returns(LabeledTensor)
[all …]
Dio_ops_test.py54 self.serialized = core.LabeledTensor(
66 expected_a = core.LabeledTensor(constant_op.constant([1, 5]), ['batch'])
67 expected_b = core.LabeledTensor(
74 serialized = core.LabeledTensor(
83 expected_a = core.LabeledTensor(constant_op.constant(1), [])
84 expected_b = core.LabeledTensor(constant_op.constant([2, 3, 4]), ['x'])
Dio_ops.py79 @tc.returns(tc.Dict(string_types, core.LabeledTensor))
112 parsed[name] = core.LabeledTensor(parsed_feature, axes)
117 @tc.returns(tc.Dict(string_types, core.LabeledTensor))
147 parsed[name] = core.LabeledTensor(parsed_feature, features[name].axes)
152 @tc.returns(core.LabeledTensor)
178 return core.LabeledTensor(tensor, axes)
Dnn_test.py32 self.original_lt = core.LabeledTensor([0.0, 0.5, 1.0], self.axes)
48 golden_lt = core.LabeledTensor(golden_tensor, self.axes)
67 golden_lt = core.LabeledTensor(golden_tensor, self.axes)
Dsugar.py63 @tc.returns(core.LabeledTensor)
100 @tc.returns(core.LabeledTensor)
Dsugar_test.py37 self.small_lt = core.LabeledTensor(constant_op.constant([1]), [('x', 1)])
69 self.masked_image_lt = core.LabeledTensor(tensor, axes)
/external/tensorflow/tensorflow/contrib/labeled_tensor/
DREADME.md3 LabeledTensor is a library for adding semantically meaningful dimension and
6 LabeledTensor was inspired by [xarray](http://xarray.pydata.org) and
11 `LabeledTensor` is an immutable object consisting of two components:
35 image = lt.LabeledTensor(raw_image, axes)
37 weights = lt.LabeledTensor(tf.constant([0.1, 0.3, 0.6]),
53 LabeledTensor has wrappers for [quite a
D__init__.py32 LabeledTensor = _core.LabeledTensor variable