Searched refs:step_ids (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/contrib/seq2seq/ops/ |
| D | beam_search_ops.cc | 33 ShapeHandle step_ids, parent_ids, max_sequence_lengths, end_token; in __anon802694180102() local
38 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &step_ids)); in __anon802694180102()
42 TF_RETURN_IF_ERROR(c->Merge(step_ids, parent_ids, &step_ids)); in __anon802694180102()
43 DimensionHandle batch_size = c->Dim(step_ids, 1); in __anon802694180102()
46 ShapeHandle step_ids_prefix = c->Matrix(c->Dim(step_ids, 0), batch_size); in __anon802694180102()
47 TF_RETURN_IF_ERROR(c->MergePrefix(step_ids, step_ids_prefix, &step_ids, in __anon802694180102()
50 c->set_output(0, step_ids); in __anon802694180102()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/ |
| D | beam_search_ops_test.py | 40 step_ids = _transpose_batch_time(
48 step_ids=step_ids,
59 step_ids = _transpose_batch_time(
68 step_ids=step_ids,
82 step_ids = _transpose_batch_time(
91 step_ids=step_ids,
105 step_ids = np.random.randint(
111 step_ids=step_ids.astype(np.int32),
|
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| D | beam_search_ops.cc | 50 const Tensor& step_ids = ctx->input(0); in Compute() local
54 const TensorShape& step_ids_shape = step_ids.shape(); in Compute()
83 typename TTypes<T, 3>::ConstTensor step_ids_t(step_ids.tensor<T, 3>()); in Compute()
108 TTypes<int32, 3>::ConstTensor step_ids, in operator ()()
128 step_ids(max_seq_len_b - 1, batch, beam); in operator ()()
138 beams(level, batch, beam) = step_ids(level, batch, parent); in operator ()()
174 typename TTypes<T, 3>::ConstTensor step_ids, \
|
| D | beam_search_ops_gpu.cu.cc | 30 const int32 beam_width, const T* step_ids, in GatherTreeOpKernel() argument
47 beams[initial_beam_ix] = ldg(step_ids + initial_beam_ix); in GatherTreeOpKernel()
58 beams[level_beam_ix] = ldg(step_ids + level_parent_ix); in GatherTreeOpKernel()
83 typename TTypes<T, 3>::ConstTensor step_ids, in operator ()()
96 d.stream(), batch_size, max_time, beam_width, step_ids.data(), in operator ()()
|
| D | beam_search_ops.h | 31 typename TTypes<T, 3>::ConstTensor step_ids,
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 543 // a common collective instance to coordinate using the same step_ids.
553 // Next valid step_ids for one or more graph_keys.
|
| D | config.proto | 495 // ops and must synchronize step_ids with any other graph with this
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 155 step_ids=beam_ids,
|