Searched refs:is_nested_list (Results 1 – 3 of 3) sorted by relevance
50 bool is_nested_list; in Compile() local51 OP_REQUIRES_OK(ctx, IsNestedTensorList(ctx->Input(0), &is_nested_list)); in Compile()52 OP_REQUIRES(ctx, !is_nested_list, in Compile()
34 Status IsNestedTensorList(xla::XlaOp list, bool* is_nested_list);
121 Status IsNestedTensorList(xla::XlaOp list, bool* is_nested_list) { in IsNestedTensorList() argument128 *is_nested_list = (xla::ShapeUtil::TupleElementCount(list_shape) > 2); in IsNestedTensorList()