Searched refs:pos_shape (Results 1 – 2 of 2) sorted by relevance
130 ShapeHandle pos_shape = c->input(1); in __anon5bd2fe870302() local134 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()
44 const TensorShape& pos_shape = pos_tensor.shape(); in Compute() local46 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()