| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| beam_search_ops_test.py | 42 parent_ids = _transpose_batch_time(
49 parent_ids=parent_ids,
61 parent_ids = _transpose_batch_time(
69 parent_ids=parent_ids,
84 parent_ids = _transpose_batch_time(
92 parent_ids=parent_ids,
107 parent_ids = np.random.randint
[all...] |
| beam_search_decoder_test.py | 55 parent_ids = np.array(
68 parent_ids,
83 parent_ids = np.array(
93 parent_ids = ops.convert_to_tensor(
94 parent_ids, dtype=dtypes.int32)
119 array, parent_ids, sequence_length)
148 parent_ids = np.array(
168 parent_ids = ops.convert_to_tensor(
169 parent_ids, dtype=dtypes.int32)
174 array, parent_ids, sequence_length
[all...] |
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| beam_search_ops_gpu.cu.cc | 31 const T* parent_ids,
48 int32 parent = ldg(parent_ids + initial_beam_ix);
59 parent = ldg(parent_ids + level_parent_ix);
84 typename TTypes<T, 3>::ConstTensor parent_ids,
87 const int32 max_time = parent_ids.dimension(0);
88 const int32 batch_size = parent_ids.dimension(1);
89 const int32 beam_width = parent_ids.dimension(2);
97 parent_ids.data(), max_sequence_length.data(), end_token,
|
| beam_search_ops.h | 32 typename TTypes<T, 3>::ConstTensor parent_ids,
|
| beam_search_ops.cc | 51 const Tensor& parent_ids = ctx->input(1); variable
68 ctx, step_ids_shape == parent_ids.shape(),
70 "step_ids.shape must match parent_ids.shape. but shapes are: ",
72 parent_ids.shape().DebugString()));
84 typename TTypes<T, 3>::ConstTensor parent_ids_t(parent_ids.tensor<T, 3>());
109 TTypes<int32, 3>::ConstTensor parent_ids,
112 const int32 max_time = parent_ids.dimension(0);
113 const int32 batch_size = parent_ids.dimension(1);
114 const int32 beam_width = parent_ids.dimension(2);
129 int32 parent = parent_ids(max_seq_len_b - 1, batch, beam)
[all...] |
| /external/tensorflow/tensorflow/contrib/seq2seq/ops/ |
| beam_search_ops.cc | 27 .Input("parent_ids: T")
33 ShapeHandle step_ids, parent_ids, max_sequence_lengths, end_token;
35 // step_ids, parent_ids, and output are all shaped:
39 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 3, &parent_ids));
42 TF_RETURN_IF_ERROR(c->Merge(step_ids, parent_ids, &step_ids));
66 parent_ids: `[max_time, batch_size, beam_width]`.
|
| /external/u-boot/drivers/clk/uniphier/ |
| clk-uniphier-mio.c | 24 .parent_ids = { \
|
| clk-uniphier.h | 33 u8 parent_ids[UNIPHIER_CLK_MUX_MAX_PARENTS]; member in struct:uniphier_clk_mux_data
|
| clk-uniphier-core.c | 46 if (mux->parent_ids[i] != id)
69 return mux->parent_ids[i];
202 parent_id = data->data.mux.parent_ids[i];
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| beam_search_decoder.py | 61 ("scores", "predicted_ids", "parent_ids"))):
129 def gather_tree_from_array(t, parent_ids, sequence_length):
136 parent_ids: The parent ids of shape `[max_time, batch_size, beam_width]`.
141 `t` and where beams are sorted in each `Tensor` according to `parent_ids`.
143 max_time = parent_ids.shape.dims[0].value or array_ops.shape(parent_ids)[0]
144 batch_size = parent_ids.shape.dims[1].value or array_ops.shape(parent_ids)[1]
145 beam_width = parent_ids.shape.dims[2].value or array_ops.shape(parent_ids)[2
[all...] |
