HomeSort by relevance Sort by last modified time
    Searched refs:step_ids (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/contrib/seq2seq/ops/
beam_search_ops.cc 26 .Input("step_ids: T")
33 ShapeHandle step_ids, parent_ids, max_sequence_lengths, end_token;
35 // step_ids, parent_ids, and output are all shaped:
38 TF_RETURN_IF_ERROR(c->WithRank(c->input(0), 3, &step_ids));
42 TF_RETURN_IF_ERROR(c->Merge(step_ids, parent_ids, &step_ids));
43 DimensionHandle batch_size = c->Dim(step_ids, 1);
46 ShapeHandle step_ids_prefix = c->Matrix(c->Dim(step_ids, 0), batch_size);
47 TF_RETURN_IF_ERROR(c->MergePrefix(step_ids, step_ids_prefix, &step_ids,
    [all...]
  /external/tensorflow/tensorflow/contrib/seq2seq/python/kernel_tests/
beam_search_ops_test.py 40 step_ids = _transpose_batch_time(
48 step_ids=step_ids,
59 step_ids = _transpose_batch_time(
66 step_ids=step_ids,
83 step_ids = _transpose_batch_time(
92 step_ids=step_ids,
107 step_ids = np.random.randint
    [all...]
  /external/tensorflow/tensorflow/contrib/seq2seq/kernels/
beam_search_ops.h 32 typename TTypes<T, 3>::ConstTensor step_ids,
beam_search_ops_gpu.cu.cc 30 const int32 beam_width, const T* step_ids,
47 beams[initial_beam_ix] = ldg(step_ids + initial_beam_ix);
58 beams[level_beam_ix] = ldg(step_ids + level_parent_ix);
83 typename TTypes<T, 3>::ConstTensor step_ids,
98 step_ids.data(),
beam_search_ops.cc 50 const Tensor& step_ids = ctx->input(0); variable
54 const TensorShape& step_ids_shape = step_ids.shape();
57 errors::InvalidArgument("step_ids must be a 3-tensor, saw shape: ",
70 "step_ids.shape must match parent_ids.shape. but shapes are: ",
76 errors::InvalidArgument("batch size dimensions step_ids.shape[1] and "
83 typename TTypes<T, 3>::ConstTensor step_ids_t(step_ids.tensor<T, 3>());
108 TTypes<int32, 3>::ConstTensor step_ids,
128 step_ids(max_seq_len_b - 1, batch, beam);
138 beams(level, batch, beam) = step_ids(level, batch, parent);
174 typename TTypes<T, 3>::ConstTensor step_ids, \
    [all...]

Completed in 75 milliseconds