Home
last modified time | relevance | path

Searched refs:pos_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dstring_ops.cc130 ShapeHandle pos_shape = c->input(1); in __anon5bd2fe870302() local
134 TF_RETURN_IF_ERROR(c->WithRank(pos_shape, c->Rank(len_shape), &unused)); in __anon5bd2fe870302()
136 for (int32 i = 0; i < c->Rank(pos_shape); ++i) { in __anon5bd2fe870302()
137 DimensionHandle pos_dim = c->Dim(pos_shape, i); in __anon5bd2fe870302()
141 "pos and len shapes must match: ", c->DebugString(pos_shape), in __anon5bd2fe870302()
/external/tensorflow/tensorflow/core/kernels/
Dsubstr_op.cc44 const TensorShape& pos_shape = pos_tensor.shape(); in Compute() local
46 bool is_scalar = TensorShapeUtils::IsScalar(pos_shape); in Compute()
48 if (is_scalar || input_shape == pos_shape) { in Compute()
96 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(pos_shape)); in Compute()
100 " vs. ", pos_shape.DebugString())); in Compute()