Searched refs:CustomTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/util/ |
| D | dispatch_test.py | 30 class CustomTensor(object): class
52 @dispatch.dispatch_for_types(gen_math_ops.add, CustomTensor)
54 return CustomTensor(gen_math_ops.add(x.tensor, y.tensor, name),
60 x = CustomTensor([1, 2, 3], 2.0)
61 y = CustomTensor([7, 8, 2], 0.0)
83 @dispatch.dispatch_for_types(test_op, CustomTensor)
85 return CustomTensor(test_op(x.tensor, y.tensor, z.tensor),
88 x = CustomTensor([1, 2, 3], 0.2)
89 y = CustomTensor([7, 8, 2], 0.4)
90 z = CustomTensor([0, 1, 2], 0.6)
[all …]
|