Searched refs:context_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | sequence_feature_column_test.py | 56 context_input = ops.convert_to_tensor(np.arange(10).reshape(2, 5))
58 context_input = math_ops.cast(context_input, dtype=dtypes.float32)
59 input_layer = sfc.concatenate_context_input(context_input, seq_input)
76 context_input = ops.convert_to_tensor(np.arange(100).reshape(10, 10))
78 context_input = math_ops.cast(context_input, dtype=dtypes.float32)
80 sfc.concatenate_context_input(context_input, seq_input)
89 context_input = ops.convert_to_tensor(context_input_arg)
92 context_input = math_ops.cast(context_input, dtype=dtypes.float32)
94 sfc.concatenate_context_input(context_input, seq_input)
98 context_input = ops.convert_to_tensor(np.arange(100).reshape(10, 10))
[all …]
|
| D | sequence_feature_column.py | 41 def concatenate_context_input(context_input, sequence_input): argument
72 context_input,
75 data=[array_ops.shape(context_input)])
77 context_input,
80 context_input.dtype))
85 array_ops.expand_dims(context_input, 1),
|