Searched refs:dim_indices (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/grid_rnn/python/ops/ |
| D | grid_rnn_cell.py | 611 def _propagate(dim_indices, conf, cells, c_prev, m_prev, new_output, new_state, argument
615 if len(dim_indices) == 0:
637 for i in dim_indices:
646 if conf.tied and not (first_call and i == dim_indices[0]):
665 if conf.tied and not (first_call and i == dim_indices[0]):
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 251 std::vector<int64> PickDims(gtl::ArraySlice<int64> dim_indices) const { in PickDims() argument
252 std::vector<int64> res(dim_indices.size()); in PickDims()
253 for (size_t i = 0; i < dim_indices.size(); ++i) { in PickDims()
254 res[i] = shape_[dim_indices[i]]; in PickDims()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3504 dim_indices = gen_math_ops._range(start, dim_value, step)
3505 dim_indices *= accum_dim_value
3508 batch_indices += reshape(dim_indices, dim_shape)
|