Searched refs:pos_shape (Results 1 – 2 of 2) sorted by relevance
239 ShapeHandle pos_shape = c->input(1); in __anonb4ab91930702() local243 TF_RETURN_IF_ERROR(c->WithRank(pos_shape, c->Rank(len_shape), &unused)); in __anonb4ab91930702()245 for (int32 i = 0; i < c->Rank(pos_shape); ++i) { in __anonb4ab91930702()246 DimensionHandle pos_dim = c->Dim(pos_shape, i); in __anonb4ab91930702()250 "pos and len shapes must match: ", c->DebugString(pos_shape), in __anonb4ab91930702()
53 const TensorShape& pos_shape = pos_tensor.shape(); in Compute() local55 bool is_scalar = TensorShapeUtils::IsScalar(pos_shape); in Compute()57 if (is_scalar || input_shape == pos_shape) { in Compute()131 BCast bcast(BCast::FromShape(input_shape), BCast::FromShape(pos_shape)); in Compute()135 " vs. ", pos_shape.DebugString())); in Compute()